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/core.meta.slang | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'source/slang/core.meta.slang') diff --git a/source/slang/core.meta.slang b/source/slang/core.meta.slang index 956a5b29a..e989e4ffa 100644 --- a/source/slang/core.meta.slang +++ b/source/slang/core.meta.slang @@ -2263,6 +2263,12 @@ attribute_syntax [CudaHost] : CudaHostAttribute; __attributeTarget(FuncDecl) attribute_syntax [CudaKernel] : CudaKernelAttribute; +__attributeTarget(FuncDecl) +attribute_syntax[AutoPyBindCUDA] : AutoPyBindCudaAttribute; + +__attributeTarget(AggTypeDecl) +attribute_syntax [PyExport(name: String)] : PyExportAttribute; + __attributeTarget(InterfaceDecl) attribute_syntax [COM(guid: String)] : ComInterfaceAttribute; -- cgit v1.2.3