fix hip

xinhaoc · Jan 19, 2024 · c162d4c · c162d4c
1 parent fb91122
commit c162d4c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/ops/fused.cpp b/src/ops/fused.cpp
@@ -205,9 +205,7 @@ __host__ void FusedOp::forward_task(Task const *task,
         assert(fused->op_num_outputs[op] == 1);
         DropoutMeta *m = (DropoutMeta *)metas->meta[op];
         Kernels::Dropout::forward_kernel_wrapper(
-            m,
-            my_input_accessor[0].get_float_ptr(),
-            my_output_accessor[0].get_float_ptr());
+            m, my_input_accessor[0], my_output_accessor[0]);
         break;
       }
       case OP_LINEAR: {