Skip to content

Commit d1122d5

Browse files
committed
cuda: skip compilation when overriding PTX file
1 parent 72f1191 commit d1122d5

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

src/runtime/cuda/cuda_runtime_program.c

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,12 @@ static bool emit_cuda_c_code(CudaKernel* spec) {
3636
}
3737

3838
static bool cuda_c_to_ptx(CudaKernel* kernel) {
39+
String override_file = getenv("SHADY_OVERRIDE_PTX");
40+
if (override_file) {
41+
read_file(override_file, &kernel->ptx_size, &kernel->ptx);
42+
return true;
43+
}
44+
3945
nvrtcProgram program;
4046
CHECK_NVRTC(nvrtcCreateProgram(&program, kernel->cuda_code, kernel->key.entry_point, 0, NULL, NULL), return false);
4147

@@ -68,11 +74,6 @@ static bool cuda_c_to_ptx(CudaKernel* kernel) {
6874
CHECK_NVRTC(nvrtcGetPTX(program, kernel->ptx), return false);
6975
CHECK_NVRTC(nvrtcDestroyProgram(&program), return false);
7076

71-
String override_file = getenv("SHADY_OVERRIDE_PTX");
72-
if (override_file) {
73-
read_file(override_file, &kernel->ptx_size, &kernel->ptx);
74-
}
75-
7677
if (get_log_level() <= DEBUG)
7778
write_file("cuda_dump.ptx", kernel->ptx_size - 1, kernel->ptx);
7879

0 commit comments

Comments
 (0)