FM / cuda

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/cuda/

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/cuda

Name	Size	Mode	Actions
detail/	-	0755	rm
ATenCUDAGeneral.h	190	0644	edit dl rm
cub.cuh	11594	0644	edit dl rm
CUDAApplyUtils.cuh	21290	0644	edit dl rm
CUDABlas.h	11001	0644	edit dl rm
CUDAConfig.h	583	0644	edit dl rm
CUDAContext.h	2249	0644	edit dl rm
CUDADevice.h	352	0644	edit dl rm
CUDAEvent.h	5738	0644	edit dl rm
CUDAGraph.h	2437	0644	edit dl rm
CUDAGraphsUtils.cuh	2013	0644	edit dl rm
CUDASolver.h	22878	0644	edit dl rm
CUDATensorMethods.cuh	270	0644	edit dl rm
CUDAUtils.h	428	0644	edit dl rm
DeviceUtils.cuh	2991	0644	edit dl rm
Exceptions.h	7387	0644	edit dl rm
NumericLimits.cuh	5214	0644	edit dl rm
PinnedMemoryAllocator.h	167	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/cuda/CUDAConfig.h (583B)

#pragma once // Test these using #if AT_CUDNN_ENABLED(), not #ifdef, so that it's // obvious if you forgot to include Config.h // c.f. https://stackoverflow.com/questions/33759787/generating-an-error-if-checked-boolean-macro-is-not-defined // // NB: This header MUST NOT be included from other headers; it should // only be included from C++ files. #define AT_CUDNN_ENABLED() 1 #define AT_ROCM_ENABLED() 0 #define NVCC_FLAGS_EXTRA "-gencode;arch=compute_37,code=sm_37;-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_70,code=sm_70"