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-compiler.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'source/slang/slang-compiler.cpp') diff --git a/source/slang/slang-compiler.cpp b/source/slang/slang-compiler.cpp index 8cbe12ef0..ff38bbbde 100644 --- a/source/slang/slang-compiler.cpp +++ b/source/slang/slang-compiler.cpp @@ -470,6 +470,7 @@ namespace Slang case CodeGenTarget::CUDASource: case CodeGenTarget::CPPSource: case CodeGenTarget::HostCPPSource: + case CodeGenTarget::PyTorchCppBinding: case CodeGenTarget::CSource: { return PassThroughMode::None; @@ -1570,6 +1571,7 @@ namespace Slang case CodeGenTarget::CUDASource: case CodeGenTarget::CPPSource: case CodeGenTarget::HostCPPSource: + case CodeGenTarget::PyTorchCppBinding: case CodeGenTarget::CSource: { RefPtr extensionTracker = _newExtensionTracker(target); -- cgit v1.2.3