Skip to content

Commit e59f9f7

Browse files
committed
Add Battlemage G21 arhitecture to w/a
1 parent ada070e commit e59f9f7

File tree

2 files changed

+30
-6
lines changed

2 files changed

+30
-6
lines changed

dpnp/backend/extensions/blas/gemm.cpp

Lines changed: 25 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -328,14 +328,37 @@ bool _is_lnl_arl_architecture(sycl::device &dev)
328328
#if !defined(USE_ONEMKL_CUBLAS)
329329
if (dev.ext_oneapi_architecture_is(
330330
sycl::ext::oneapi::experimental::architecture::
331-
intel_gpu_20_4_4)) /* Lunar Lake */
331+
intel_gpu_lnl_m)) /* Lunar Lake */
332332
{
333+
std::cout << "_is_lnl_arl_architecture: running on Lunar Lake" << std::endl;
334+
return true;
335+
}
336+
if (dev.ext_oneapi_architecture_is(
337+
sycl::ext::oneapi::experimental::architecture::
338+
intel_gpu_bmg_g21)) /* Battlemage G21 */
339+
{
340+
std::cout << "_is_lnl_arl_architecture: running on Battlemage G21" << std::endl;
341+
return true;
342+
}
343+
else if (dev.ext_oneapi_architecture_is(
344+
sycl::ext::oneapi::experimental::architecture::
345+
intel_gpu_arl_h)) /* Arrow Lake H */
346+
{
347+
std::cout << "_is_lnl_arl_architecture: running on Arrow Lake H" << std::endl;
348+
return true;
349+
}
350+
else if (dev.ext_oneapi_architecture_is(
351+
sycl::ext::oneapi::experimental::architecture::
352+
intel_gpu_arl_u)) /* Arrow Lake U */
353+
{
354+
std::cout << "_is_lnl_arl_architecture: running on Arrow Lake U" << std::endl;
333355
return true;
334356
}
335357
else if (dev.ext_oneapi_architecture_is(
336358
sycl::ext::oneapi::experimental::architecture::
337-
intel_gpu_12_74_4)) /* Arrow Lake */
359+
intel_gpu_arl_s)) /* Arrow Lake S */
338360
{
361+
std::cout << "_is_lnl_arl_architecture: running on Arrow Lake S" << std::endl;
339362
return true;
340363
}
341364
#endif // !defined(USE_ONEMKL_CUBLAS)

dpnp/dpnp_utils/dpnp_utils_linearalgebra.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -894,10 +894,11 @@ def dpnp_matmul(
894894
)
895895
_manager.add_event_pair(ht_ev, gemv_ev)
896896
elif call_flag == "gemm":
897-
# MKLD-17976: due to known issue in OneMKL on Lunar Lake and
898-
# Arrow Lake architectures, it forces to implement a temporary
899-
# workaround with extra copying of an input array in case when
900-
# it has a small size and non-zero offset
897+
# MKLD-17976: due to known issue in OneMKL on Lunar Lake,
898+
# Battlemage G21 and Arrow Lake architectures, it forces
899+
# to implement a temporary workaround with extra copying
900+
# of an input array in case when it has a small size and
901+
# non-zero offset
901902
# TODO: remove the workaround once OneMKL issue is resolved
902903
if bi._is_lnl_arl_architecture(exec_q.get_sycl_device()):
903904
def _need_to_copy(a):

0 commit comments

Comments
 (0)