ATen
ATen copied to clipboard
zdevito
ATen: A TENsor library for C++11
The CUDNN function `cudnnFindConvolutionForwardAlgorithmEx` (used in https://github.com/zdevito/ATen/blob/master/aten/src/ATen/native/cudnn/Conv.cpp#L481) searches an optimal algorithm satisfying the provided workspace constraint by varying the algorithm type but also by changing the math type. The current...
Hi, I have a neural network which expects input size as (batch_size, 1, 112, 112). If batch size is 1, it takes about 2ms for a forward pass. If I...
Currently if you want to create a new Tensor from specifications residing on gpu a sync is needed (imho): It would be greate if sthg like the following could work:...
Super sorry if this is obvious, but -- how do I copy a tensor from CPU -> GPU and vice versa? I've been looking through the documentation, and can't seem...
My machine has cuda install at: /usr/local/cuda; it's fine when I build with cuda. However, if I don't want to build it with cuda ```cmake .. -DNO_CUDA=1``` I get following...
How to convert Type of Aten? from kByte to kFloat?
May I initializee at::Tensor from std:;vector
I have tried to build ATen. and CMake Error at src/ATen/CMakeLists.txt:407 (add_subdirectory): add_subdirectory given source "E:/ATen/aten/src/ATen/../../../third_party/cpuinfo" which is not an existing directory. CMake Error at src/ATen/CMakeLists.txt:412 (set_property): set_property could not...
I'm trying to call function from inside ATen and don't know how to get cusparseHandle_t handle. Could you help me?
