Skip to content

GitLab

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

Remove casting warning

parent aece8f87
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
nunchaku/csrc/gemm.h
View file @ 618f3078
...@@ -67,7 +67,7 @@ public: ...@@ -67,7 +67,7 @@ public:
const int offset = ((bm * (K / WARP_K) + bn) * NUM_WARPS + warpId) * WARP_M_TILES * WARP_SIZE * 4; const int offset = ((bm * (K / WARP_K) + bn) * NUM_WARPS + warpId) * WARP_M_TILES * WARP_SIZE * 4;
for (int i = 0; i < 16; i++) { 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]; uint32_t val = x.data_ptr<uint32_t>()[offset + i];
ss << "{"; ss << "{";
for (int j = 0; j < 8; j++) { for (int j = 0; j < 8; j++) {
...@@ -110,4 +110,4 @@ public: ...@@ -110,4 +110,4 @@ public:
spdlog::debug("ascales = {}", dumpTensorBF16(ascales)); 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