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 --- tests/autodiff/reverse-loop-diff-only-2.slang | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/autodiff/reverse-loop-diff-only-2.slang') diff --git a/tests/autodiff/reverse-loop-diff-only-2.slang b/tests/autodiff/reverse-loop-diff-only-2.slang index aad405b48..2cc33ecca 100644 --- a/tests/autodiff/reverse-loop-diff-only-2.slang +++ b/tests/autodiff/reverse-loop-diff-only-2.slang @@ -32,7 +32,7 @@ float infinitesimal(float x) // Test that computeLoop's intermediates have no float sitting // around (must not cache the outvar from 'compute()') -// CHECK: struct s_bwd_computeLoop_Intermediates +// CHECK: struct s_bwd_prop_computeLoop_Intermediates // CHECK-NEXT: { // CHECK-NOT: {{[A-Za-z0-9_]+}} {{[A-Za-z0-9_]+}}[{{.*}}] // CHECK: } -- cgit v1.2.3