fix miopenOpTensor

Bob-Chen222 · Jun 5, 2024 · a2a3dd8 · a2a3dd8
1 parent 504f6c2
commit a2a3dd8
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/lib/kernels/src/hip/reduce_kernels.cpp b/lib/kernels/src/hip/reduce_kernels.cpp
@@ -87,13 +87,17 @@ void backward_kernel(hipStream_t stream,
     default:
       assert(false);
   }
-  checkCUDNN(hipdnnAddTensor(m.handle.dnn,
-                             &alpha,
-                             m.outputTensor,
-                             output_grad_ptr,
-                             &beta,
-                             m.inputTensor,
-                             input_grad_ptr));
+  checkCUDNN(miopenOpTensor(m.handle.dnn,
+                            miopenTensorOpAdd,
+                            &alpha,
+                            m.inputTensor,
+                            input_grad_ptr,
+                            &alpha,
+                            m.outputTensor,
+                            output_grad_ptr,
+                            &beta,
+                            m.inputTensor,
+                            input_grad_ptr));
 }
 
 } // namespace Reduce