Skip to content

Commit 89fa34a

Browse files
committed
cuda: added envvar override
1 parent 0a3eca0 commit 89fa34a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/runtime/cuda/cuda_runtime_program.c

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,11 @@ static bool cuda_c_to_ptx(CudaKernel* kernel) {
5555
CHECK_NVRTC(nvrtcGetPTX(program, kernel->ptx), return false);
5656
CHECK_NVRTC(nvrtcDestroyProgram(&program), return false);
5757

58+
String override_file = getenv("SHADY_OVERRIDE_PTX");
59+
if (override_file) {
60+
read_file(override_file, &kernel->ptx_size, &kernel->ptx);
61+
}
62+
5863
return true;
5964
}
6065

0 commit comments

Comments
 (0)