diff options
Diffstat (limited to 'source/slang/slang-ir-pytorch-cpp-binding.h')
| -rw-r--r-- | source/slang/slang-ir-pytorch-cpp-binding.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/source/slang/slang-ir-pytorch-cpp-binding.h b/source/slang/slang-ir-pytorch-cpp-binding.h index a761dbc03..6d022db7b 100644 --- a/source/slang/slang-ir-pytorch-cpp-binding.h +++ b/source/slang/slang-ir-pytorch-cpp-binding.h @@ -11,6 +11,7 @@ void removeTorchKernels(IRModule* module); void handleAutoBindNames(IRModule* module); void generateDerivativeWrappers(IRModule* module, DiagnosticSink* sink); void lowerBuiltinTypesForKernelEntryPoints(IRModule* module, DiagnosticSink* sink); +void removeTorchAndCUDAEntryPoints(IRModule* module); } |