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

Commit 0e5706f

Browse files
sunjiweiswiftDDEle
authored andcommitted
bugfix for update offset_x/offset_y
1 parent 18cc4e1 commit 0e5706f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

include/subgroup/tile/impl/payload_xe.hpp

+2
Original file line numberDiff line numberDiff line change
@@ -231,12 +231,14 @@ struct mem_payload_t<
231231
__XETLA_API void update_tdesc(int offset) {
232232
auto payloads_2d = payloads.xetla_format<uint32_t, num_block, 16>();
233233
if constexpr (update_dir == tdesc_update_dir::x_dir) {
234+
offset_x += offset / scale_factor;
234235
#pragma unroll
235236
for (uint32_t i = 0; i < num_block; i++) {
236237
xetla_update_tdesc_offsetx(
237238
payloads_2d.row(i), offset / int32_t(scale_factor));
238239
}
239240
} else {
241+
offset_y += offset;
240242
#pragma unroll
241243
for (uint32_t i = 0; i < num_block; i++) {
242244
xetla_update_tdesc_offsety(payloads_2d.row(i), offset);

0 commit comments

Comments
 (0)