Commit 618f3078 authored by Hyunsung Lee's avatar Hyunsung Lee Committed by Zhekai Zhang
Browse files

Remove casting warning

parent aece8f87
......@@ -67,7 +67,7 @@ public:
const int offset = ((bm * (K / WARP_K) + bn) * NUM_WARPS + warpId) * WARP_M_TILES * WARP_SIZE * 4;
for (int i = 0; i < 16; i++) {
assert(offset + i < x.numel() / 4);
assert(static_cast<size_t>(offset + i) < x.numel() / 4);
uint32_t val = x.data_ptr<uint32_t>()[offset + i];
ss << "{";
for (int j = 0; j < 8; j++) {
......@@ -110,4 +110,4 @@ public:
spdlog::debug("ascales = {}", dumpTensorBF16(ascales));
}
};
\ No newline at end of file
};
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment