I was trying to load a cpp extension with
torch.utils.cpp_extension.load() on my Jetson TX2 device with CUDA-10.2.
from torch.utils.cpp_extension import load matrix_op_cuda = load( 'matrix_op_cuda', ['./matrix_op/matrix_op_cuda.cpp', './matrix_op/matrix_op_cuda_kernel.cu'], verbose=True)
The compilation process with ninja went smoothly, but when loading the extension module, the process was killed and the log showed no more info.
I tried to load the extension again but the same thing happened.
Building extension module matrix_op_cuda... Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N) ninja: no work to do. Loading extension module matrix_op_cuda... Killed
Wonder what could be the problem?