diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/autodiff/autopybind-differentiable.slang | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/autodiff/autopybind-differentiable.slang b/tests/autodiff/autopybind-differentiable.slang index 9595a09d2..27df19b8a 100644 --- a/tests/autodiff/autopybind-differentiable.slang +++ b/tests/autodiff/autopybind-differentiable.slang @@ -12,7 +12,7 @@ void myKernel(DiffTensorView inValues, DiffTensorView outValues) { if (cudaThreadIdx().x > 0) return; - outValues.store(cudaThreadIdx().x, sin(inValues.load(cudaThreadIdx().x))); + outValues[cudaThreadIdx().x] = sin(inValues[cudaThreadIdx().x]); } // TORCH: {{^SLANG_PRELUDE_EXPORT$}} |