From e7238942ea003e134b325fa65296df8e19368e90 Mon Sep 17 00:00:00 2001
From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com>
Date: Thu, 28 Sep 2023 02:19:54 -0400
Subject: Fix `TensorView.store()` definition (#3248)

---
 source/slang/diff.meta.slang | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'source')

diff --git a/source/slang/diff.meta.slang b/source/slang/diff.meta.slang
index e635e9609..31ed5845e 100644
--- a/source/slang/diff.meta.slang
+++ b/source/slang/diff.meta.slang
@@ -105,7 +105,7 @@ struct TensorView
     void store(uint i0, uint i1, uint i2, uint i3, uint i4, T val);
 
     __generic<let N : int>
-    __target_intrinsic(cuda, "$0.store<$T1>($1)")
+    __target_intrinsic(cuda, "$0.store<$T2>($1, $2)")
     void store(vector<uint, N> index, T val);
 
     __target_intrinsic(cuda, "*($3) = atomicAdd($0.data_ptr_at<$T2>($1), $2)")
-- 
cgit v1.2.3