[FIX] Fix sycl esimd feature not support on HOST error (#2666)

intel · Apr 9, 2024 · a7c62a1 · a7c62a1
1 parent faf8925
commit a7c62a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/itex/core/kernels/gpu/xetla/fmha_backward.h b/itex/core/kernels/gpu/xetla/fmha_backward.h
@@ -87,7 +87,7 @@ struct arguments_t {
         uH(head_size),
         uF(num_queries),
         uT(num_keys),
-        sm_scale(xetla_rsqrt<accum_t>(accum_t(head_size))) {}
+        sm_scale(1. / sqrt(head_size)) {}
 };
 
 template <typename mha_policy, typename scalar_t>