Fix prelu kernel failed in bf16 since data type (#67085)

zhanglirong1999 · web-flow · commit 7f828ab41d9c · 2024-08-07T09:41:57.000+08:00
diff --git a/paddle/phi/backends/onednn/onednn_reuse.h b/paddle/phi/backends/onednn/onednn_reuse.h
@@ -1293,18 +1293,18 @@ class PReluOneDNNHandler
 
   std::shared_ptr<memory> AcquireWeightsMemoryPossiblyWithReorder(
       const DenseTensor* weights, const bool is_test) {
-    const T* weights_data = weights->data<T>();
+    const float* weights_data = weights->data<float>();
 
     // if weights are 1D, every format tag is correct, so we accept
     // format_tag::any's output and no reorder is needed
     if (weights->dims().size() == 1) {
-      return this->AcquireMemoryFromPrimitive(this->fwd_pd_->weights_desc(),
-                                              to_void_cast<T>(weights_data));
+      return this->AcquireMemoryFromPrimitive(
+          this->fwd_pd_->weights_desc(), to_void_cast<float>(weights_data));
     }
 
     return this->AcquireMemoryWithReorder(weights->mem_desc(),
                                           this->fwd_pd_->weights_desc(),
-                                          to_void_cast<T>(weights_data),
+                                          to_void_cast<float>(weights_data),
                                           is_test);
   }
 
