Skip to content
This repository has been archived by the owner on Aug 30, 2024. It is now read-only.

Commit

Permalink
fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
@luoyu-intel
luoyu-intel committed Jun 18, 2024
1 parent 7330d96 commit f4c87fb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions bestla/bestla/sycl/sycl_wrapper.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ class Launcher {
int ldb = _param.paramB.ldb;
int ldc = _param.paramC.ldc;
int m_pad = utils::padto(utils::updiv(m, GemmCore::TileM), GemmCore::WgM);
sycl::range<2> problem{m_pad, static_cast<size_t>(n) / GemmCore::TileN};
sycl::range<2> problem{static_cast<size_t>(m_pad), static_cast<size_t>(n) / GemmCore::TileN};
auto ev = q->submit([&](sycl::handler& cgh) {
sycl::local_accessor<BType, 1> slm_b(sycl::range(GemmCore::SLM_B_Size), cgh);
cgh.parallel_for(
Expand Down Expand Up @@ -148,7 +148,7 @@ class LauncherWOQ {
int ldb = _param.paramB.ldb;
int ldc = _param.paramC.ldc;
int m_pad = utils::padto(utils::updiv(m, GemmCore::TileM), GemmCore::WgM);
sycl::range<2> problem{m_pad, static_cast<size_t>(n) / GemmCore::TileN};
sycl::range<2> problem{static_cast<size_t>(m_pad), static_cast<size_t>(n) / GemmCore::TileN};
auto ev = q->submit([&](sycl::handler& cgh) {
sycl::local_accessor<BType, 1> slm_b(sycl::range(GemmCore::SLM_B_Size), cgh);
cgh.parallel_for(
Expand Down

0 comments on commit f4c87fb

Please sign in to comment.