From 739c3a7b53dc6489065fcd5e9f0a04370c5f9c8f Mon Sep 17 00:00:00 2001
From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com>
Date: Tue, 19 Sep 2023 18:51:24 -0400
Subject: Added `[AutoPyBindCUDA]` for automatic kernel binding + `[PyExport]`
 for exporting type information (#3209)

* Initial: add a DiffTensor impl

* Auto-binding and diff tensor implementations now work

* Refactored diff-tensor implementation + added py-export for struct types

* Cleanup

* Update slang-ir-pytorch-cpp-binding.cpp

* Updated test names

* Update autodiff-data-flow.slang.expected

* Add more versions of load/store & default generic args for DiffTensorView.

* Add diagnostic for default generic arg and more tests

* Add more `[AutoPyBind]` tests
---
 source/slang/slang-ir-pytorch-cpp-binding.h | 2 ++
 1 file changed, 2 insertions(+)

(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 c35b6a8eb..dd7dcc9a4 100644
--- a/source/slang/slang-ir-pytorch-cpp-binding.h
+++ b/source/slang/slang-ir-pytorch-cpp-binding.h
@@ -7,6 +7,8 @@ class DiagnosticSink;
 
 void generatePyTorchCppBinding(IRModule* module, DiagnosticSink* sink);
 void removeTorchKernels(IRModule* module);
+void handleAutoBindNames(IRModule* module);
+void generateDerivativeWrappers(IRModule* module, DiagnosticSink* sink);
 
 }
 
-- 
cgit v1.2.3