From d64ee86a3130f8eeb75d09193c38c621d7565eba Mon Sep 17 00:00:00 2001 From: Yong He Date: Sun, 26 Mar 2023 13:59:11 -0700 Subject: Add PyTorch C++ binding generation. (#2734) * Add PyTorch C++ binding generation. * fix --------- Co-authored-by: Yong He --- source/slang/slang-ir-pytorch-cpp-binding.h | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 source/slang/slang-ir-pytorch-cpp-binding.h (limited to 'source/slang/slang-ir-pytorch-cpp-binding.h') diff --git a/source/slang/slang-ir-pytorch-cpp-binding.h b/source/slang/slang-ir-pytorch-cpp-binding.h new file mode 100644 index 000000000..c35b6a8eb --- /dev/null +++ b/source/slang/slang-ir-pytorch-cpp-binding.h @@ -0,0 +1,12 @@ +#pragma once + +namespace Slang +{ +struct IRModule; +class DiagnosticSink; + +void generatePyTorchCppBinding(IRModule* module, DiagnosticSink* sink); +void removeTorchKernels(IRModule* module); + +} + -- cgit v1.2.3