Skip to content

Commit

Permalink
Removed static_cast from coordinates
Browse files Browse the repository at this point in the history
  • Loading branch information
muhammad-tanvir-1211 committed May 6, 2024
1 parent a8448ec commit 7720fc5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions include/cute/atom/copy_traits_xe.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ struct XE_2D_LD_Unpack
int H = size<0>(traits.tensor);
int W = size<1>(traits.tensor) * sizeof(typename Copy_Traits::CopyInternalType);
auto [y, x, z] = src.data().coord_;
CopyOp::copy(traits.tensor.data() + z, W, H, W, intel::coord_t{static_cast<int>(x), static_cast<int>(y)}, &*dst.data());
CopyOp::copy(traits.tensor.data() + z, W, H, W, intel::coord_t{x, y}, &*dst.data());
}

template <class GCoord, class GShape, class GStride>
Expand Down Expand Up @@ -284,7 +284,7 @@ struct XE_2D_ST_Unpack
int H = size<0>(traits.tensor);
int W = size<1>(traits.tensor) * sizeof(typename Copy_Traits::CopyInternalType);
auto [y, x, z] = dst.data().coord_;
CopyOp::copy(traits.tensor.data() + z, W, H, W, intel::coord_t{static_cast<int>(x), static_cast<int>(y)}, &*src.data());
CopyOp::copy(traits.tensor.data() + z, W, H, W, intel::coord_t{x, y}, &*src.data());
}

template <class GCoord, class GShape, class GStride>
Expand Down

0 comments on commit 7720fc5

Please sign in to comment.