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