Fixed an issue for unequal MWG and NWG and the new GEMMK == 1 kernel
parent
bca1506e87
commit
c0e41b87cb
|
@ -7,6 +7,7 @@ Development (next version)
|
|||
- Fixed an issue with conjugate transpose not being executed in certain cases for a.o. XOMATCOPY
|
||||
- Fixed an issue with AMD GPUs and the new GEMMK == 1 kernel
|
||||
- Fixed an issue with the preprocessor and the new GEMMK == 1 kernel
|
||||
- Fixed an issue for unequal MWG and NWG and the new GEMMK == 1 kernel
|
||||
- Fixed an issue for certain parameters for AXPY's 'XaxpyFaster' kernel
|
||||
- Various minor fixes and enhancements
|
||||
- Added non-BLAS routines:
|
||||
|
|
|
@ -216,9 +216,11 @@ void Xgemm<T>::GemmIndirect(const size_t m, const size_t n, const size_t k,
|
|||
kernel.SetArgument(9, static_cast<int>(c_temp_offset / db_["VWM"]));
|
||||
|
||||
// Computes the global and local thread sizes
|
||||
const auto global_divider_one = c_want_rotated_(db_["GEMMK"]) ? db_["NWG"] : db_["MWG"];
|
||||
const auto global_divider_two = c_want_rotated_(db_["GEMMK"]) ? db_["MWG"] : db_["NWG"];
|
||||
const auto global = std::vector<size_t>{
|
||||
(c_one_i * db_["MDIMC"]) / db_["MWG"],
|
||||
(c_two_i * db_["NDIMC"]) / db_["NWG"]
|
||||
(c_one_i * db_["MDIMC"]) / global_divider_one,
|
||||
(c_two_i * db_["NDIMC"]) / global_divider_two
|
||||
};
|
||||
const auto local = std::vector<size_t>{db_["MDIMC"], db_["NDIMC"]};
|
||||
|
||||
|
|
Loading…
Reference in New Issue