diff options
Diffstat (limited to 'source/slang/slang-emit.cpp')
| -rw-r--r-- | source/slang/slang-emit.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/source/slang/slang-emit.cpp b/source/slang/slang-emit.cpp index 1fa04b4be..afdd37fce 100644 --- a/source/slang/slang-emit.cpp +++ b/source/slang/slang-emit.cpp @@ -471,10 +471,13 @@ Result linkAndOptimizeIR( switch (target) { case CodeGenTarget::PyTorchCppBinding: + generateHostFunctionsForAutoBindCuda(irModule, sink); + lowerBuiltinTypesForKernelEntryPoints(irModule, sink); generatePyTorchCppBinding(irModule, sink); handleAutoBindNames(irModule); break; case CodeGenTarget::CUDASource: + lowerBuiltinTypesForKernelEntryPoints(irModule, sink); removeTorchKernels(irModule); handleAutoBindNames(irModule); break; |