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 --- tools/gfx/slang.slang | 3 ++- tools/slang-test/slang-test-main.cpp | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) (limited to 'tools') diff --git a/tools/gfx/slang.slang b/tools/gfx/slang.slang index 4250cb62e..1fd06560f 100644 --- a/tools/gfx/slang.slang +++ b/tools/gfx/slang.slang @@ -53,6 +53,7 @@ enum SlangCompileTarget SLANG_DXIL_ASM, SLANG_C_SOURCE, ///< The C language SLANG_CPP_SOURCE, ///< C++ code for shader kernels. + SLANG_CPP_PYTORCH_BINDING, SLANG_HOST_EXECUTABLE, ///< Standalone binary executable (for hosting CPU/OS) SLANG_SHADER_SHARED_LIBRARY, ///< A shared library/Dll for shader kernels (for hosting CPU/OS) SLANG_SHADER_HOST_CALLABLE, ///< A CPU target that makes the compiled shader code available to be run immediately @@ -448,4 +449,4 @@ struct SpecializationArg TypeReflection *type; } -} \ No newline at end of file +} diff --git a/tools/slang-test/slang-test-main.cpp b/tools/slang-test/slang-test-main.cpp index d2023a0f0..71f5c04bd 100644 --- a/tools/slang-test/slang-test-main.cpp +++ b/tools/slang-test/slang-test-main.cpp @@ -765,6 +765,7 @@ static PassThroughFlags _getPassThroughFlagsForTarget(SlangCompileTarget target) case SLANG_GLSL: case SLANG_C_SOURCE: case SLANG_CPP_SOURCE: + case SLANG_CPP_PYTORCH_BINDING: case SLANG_HOST_CPP_SOURCE: case SLANG_CUDA_SOURCE: { -- cgit v1.2.3