burn-tch 0.21.0

LibTorch backend for the Burn framework using the tch bindings.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

#include <iostream>
#include <stdexcept>
#include <stdint.h>
#include <stdio.h>
using namespace std;
extern "C" {
void dummy_cuda_dependency();
}

struct cublasContext;

namespace at {
namespace cuda {
cublasContext *getCurrentCUDABlasHandle();
int warp_size();
} // namespace cuda
} // namespace at
char *magma_strerror(int err);
void dummy_cuda_dependency() {
  try {
    at::cuda::getCurrentCUDABlasHandle();
    at::cuda::warp_size();
  } catch (std::exception &e) {
    if (getenv("TCH_PRINT_CUDA_INIT_ERROR") != nullptr) {
      std::cerr << "error initializing cuda: " << e.what() << std::endl;
    }
  }
}