Skip to content

Commit

Permalink
correct brgemm slm size (intel#26)
Browse files Browse the repository at this point in the history
  • Loading branch information
Jiaxingla authored Aug 28, 2023
1 parent fdcfba6 commit d644c21
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions include/group/brgemm/impl/default_xmx_xe.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -154,12 +154,8 @@ class brgemm_t<compute_policy_default_xmx<compute_attr_, perf_tuning_knob_,
public:
static constexpr uint32_t barrier_count
= enable_periodic_sync ? barrier_count_x + barrier_count_y : 0;
// current only support matA from slm
static constexpr uint32_t slm_size
= (is_local_a ? sg_tile_m * wg_size_y * k_stride * sizeof(dtype_a)
: 0)
+ (is_local_b ? sg_tile_n * wg_size_x * k_stride * sizeof(dtype_b)
: 0);

static constexpr uint32_t slm_size = 0;

static constexpr bool is_2d_block_a
= matA_payload_t::message_type == msg_type::block_2d;
Expand Down

0 comments on commit d644c21

Please sign in to comment.