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-options.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'source/slang/slang-options.cpp') diff --git a/source/slang/slang-options.cpp b/source/slang/slang-options.cpp index 714e2c99d..d30c02484 100644 --- a/source/slang/slang-options.cpp +++ b/source/slang/slang-options.cpp @@ -2193,6 +2193,7 @@ struct OptionsParser if (rawOutputs.getCount() == 0 && rawTargets.getCount() == 1 && (rawTargets[0].format == CodeGenTarget::HostCPPSource || + rawTargets[0].format == CodeGenTarget::PyTorchCppBinding || rawTargets[0].format == CodeGenTarget::CUDASource || ArtifactDescUtil::makeDescForCompileTarget(asExternal(rawTargets[0].format)).kind == ArtifactKind::HostCallable)) { @@ -2258,7 +2259,7 @@ struct OptionsParser case CodeGenTarget::ShaderHostCallable: case CodeGenTarget::HostExecutable: case CodeGenTarget::ShaderSharedLibrary: - + case CodeGenTarget::PyTorchCppBinding: case CodeGenTarget::DXIL: rawOutput.isWholeProgram = true; -- cgit v1.2.3