Skip to content

Commit b112fbc

Browse files
authored
Fix accuracy issue related to hgemm_gelu implementation (#354)
1 parent 12429d1 commit b112fbc

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

xla/service/gpu/xetla/gemm/hgemm_impl.h

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -328,8 +328,7 @@ inline void hgemm_gelu(sycl::queue& queue, scalar_t* out, const scalar_t* a,
328328
periodic_sync_interval>::gemm;
329329
using epilogue_t = group::epilogue_t<
330330
xetla::group::epilogue_policy_tile_op<
331-
xetla::subgroup::chained_tile_op_t<
332-
epilogue_impl::bias_op_t<data_type_bias>>,
331+
xetla::subgroup::chained_tile_op_t<subgroup::gelu_fwd_op_t>,
333332
gpu_arch::Xe>,
334333
tile_shape,
335334
mem_desc_t<scalar_t, mem_layout::row_major, mem_space::global>>;

0 commit comments

Comments
 (0)