Skip to content

Commit 45d26b3

Browse files
committed
Fixed constants of different size on linux and windows
Signed-off-by: Gregory Shimansky <[email protected]>
1 parent 16c3bb2 commit 45d26b3

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

third_party/intel/lib/TritonIntelGPUTransforms/MatchTargetSize.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -722,9 +722,10 @@ MatchTargetSizePass::getSubOpSize(RankedTensorType type,
722722
if (isa<ttgi::WarpEncodingAttr>(layout)) {
723723
// 32 = 2 * 16(subgroupSize) which is for large load/store
724724
// max 2d block prefetch width is 16 for 32-bit datatype
725-
subSize[1] = std::min(sizeInBits == 32 ? 16LL : 32LL, shape[1]);
725+
subSize[1] = std::min(static_cast<int64_t>(sizeInBits == 32 ? 16L : 32L),
726+
shape[1]);
726727
// max 2d block load height is 32
727-
subSize[0] = std::min(32LL, shape[0]);
728+
subSize[0] = std::min(static_cast<int64_t>(32L), shape[0]);
728729
} else if (auto dotLayout = dyn_cast<ttg::DotOperandEncodingAttr>(layout)) {
729730
const TargetArchNativeSizes::BlockMemShape &memShape =
730731
nativeSizes.getBlockMemShape(sizeInBits);

0 commit comments

Comments
 (0)