frozenleaves/oneDNN
1
0
Fork 0
mirror of https://github.com/uxlfoundation/oneDNN.git synced 2025-10-20 10:03:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

xe: sdpa: add check for host scalar fetch

Browse Source
This commit is contained in:
Kassen, Andrew
2025-10-09 11:39:43 -07:00
committed by Andy Kassen
parent f8ef3008b1
commit ff30885a40
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/gpu/intel/sdpa/micro.cpp
View File

@ -761,8 +761,10 @@ status_t micro_t::execute(const exec_ctx_t &ctx) const {
if (pd()->with_host_scale()) {
auto scalar_storage = utils::downcast<
const dnnl::impl::host_scalar_memory_storage_t *>(&scale);
scalar_storage->get_scalar_value(
auto status = scalar_storage->get_scalar_value(
&scalar_scale, scale_mdw.data_type_size());
assert(status == status::success);
if (status != status::success) return status;
scalar_scale = dnnl::impl::cpu::io::load_float_value(
pd()->scale_md()->data_type, &scalar_scale, 0);
inv_scalar_scale = 1. / scalar_scale;