This commit is contained in:
Ritwik Das 2022-05-13 15:49:10 -07:00
Родитель 4872459307
Коммит d1b7aaaa84
1 изменённых файлов: 1 добавлений и 4 удалений

Просмотреть файл

@ -46,7 +46,7 @@ def compile_cuda_program(cuda_src_path: pathlib.Path, func_name):
b'--gpu-architecture=compute_86', b'--gpu-architecture=compute_86',
b'--ptxas-options=--warn-on-spills', # https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#options-for-passing-specific-phase-options-ptxas-options b'--ptxas-options=--warn-on-spills', # https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#options-for-passing-specific-phase-options-ptxas-options
b'-use_fast_math', b'-use_fast_math',
b'--include-path=/usr/local/cuda-11.6/targets/x86_64-linux/include/', b'--include-path=/usr/local/cuda-11.6/include/',
b'-std=c++17', b'-std=c++17',
b'-default-device', b'-default-device',
#b'--restrict', #b'--restrict',
@ -74,9 +74,6 @@ def compile_cuda_program(cuda_src_path: pathlib.Path, func_name):
ptx = b" " * ptxSize ptx = b" " * ptxSize
err = nvrtc.nvrtcGetPTX(prog, ptx) err = nvrtc.nvrtcGetPTX(prog, ptx)
# prog = Program(src=src, name=func_name)
# ptx = prog.compile(opts)
return ptx return ptx