torch-build 0.1.0

Utilities to link libtorch FFI interface.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

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

struct cublasContext;

namespace at {
    namespace cuda {
        cublasContext* getCurrentCUDABlasHandle();
        int warp_size();
    }
}
char * magma_strerror(int err);
void dummy_cuda_dependency() {
    at::cuda::getCurrentCUDABlasHandle();
    at::cuda::warp_size();
}