From fba316f0e7dacc7f93bee3a95fb93b2ab02bdd80 Mon Sep 17 00:00:00 2001
From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com>
Date: Thu, 13 Jun 2024 17:30:16 -0400
Subject: Remove `IRHLSLExportDecoration` and `IRKeepAliveDecoration` for
 non-CUDA/Torch targets (#4364)

* Remove `IRHLSLExportDecoration` and `IRKeepAliveDecoration` for non-CUDA/Torch targets

* Update hlsl-torch-cross-compile.slang
---
 source/slang/slang-ir-pytorch-cpp-binding.h | 1 +
 1 file changed, 1 insertion(+)

(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
index a761dbc03..6d022db7b 100644
--- a/source/slang/slang-ir-pytorch-cpp-binding.h
+++ b/source/slang/slang-ir-pytorch-cpp-binding.h
@@ -11,6 +11,7 @@ void removeTorchKernels(IRModule* module);
 void handleAutoBindNames(IRModule* module);
 void generateDerivativeWrappers(IRModule* module, DiagnosticSink* sink);
 void lowerBuiltinTypesForKernelEntryPoints(IRModule* module, DiagnosticSink* sink);
+void removeTorchAndCUDAEntryPoints(IRModule* module);
 
 }
 
-- 
cgit v1.2.3