diff options
| author | Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> | 2023-09-28 02:19:54 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-27 23:19:54 -0700 |
| commit | e7238942ea003e134b325fa65296df8e19368e90 (patch) | |
| tree | 1bca4743d8fba2faf1474a626bcf65f11e222867 /source/slang/diff.meta.slang | |
| parent | e698b4ae92b996ac54d023c831170a5467b2d1a0 (diff) | |
Fix `TensorView.store()` definition (#3248)
Diffstat (limited to 'source/slang/diff.meta.slang')
| -rw-r--r-- | source/slang/diff.meta.slang | 2 |
1 files changed, 1 insertions, 1 deletions
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)") |