зеркало из https://github.com/microsoft/hat.git
Address PR comments
This commit is contained in:
Родитель
4872459307
Коммит
d1b7aaaa84
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|
Загрузка…
Ссылка в новой задаче