Skip to content

Commit 10ca479

Browse files
committed
fix build warnings
1 parent 46dd88c commit 10ca479

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

onnxruntime/contrib_ops/cuda/llm/gemm_profiler.h

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ void GemmPluginProfiler<Config, RunnerPtr, GemmIdType, GemmIdHashType>::profileT
330330
mDims = dims;
331331
mHasWeightOnlyCudaKernel = hasWeightOnlyCudaKernel;
332332

333-
int const maxM = std::min(nextPowerOfTwo(dims.maxM), getMaxProfileM());
333+
int const maxM = std::min(nextPowerOfTwo(static_cast<int>(dims.maxM)), getMaxProfileM());
334334

335335
size_t workspace_bytes = computeTmpSize(maxM, dims.n, dims.k);
336336

@@ -374,8 +374,8 @@ void GemmPluginProfiler<Config, RunnerPtr, GemmIdType, GemmIdHashType>::profileT
374374

375375
// Profile the (possibly reduced) set of M buckets. Any unprofiled runtime M is handled
376376
// later by lazy single-bucket profiling in getBestConfigOrProfile.
377-
for (int m : getProfileMBuckets(dims.minM, maxM, hasWeightOnlyCudaKernel)) {
378-
profileTactics(m, dims.n, dims.k);
377+
for (int m : getProfileMBuckets(static_cast<int>(dims.minM), maxM, hasWeightOnlyCudaKernel)) {
378+
profileTactics(m, static_cast<int>(dims.n), static_cast<int>(dims.k));
379379
}
380380

381381
if (isAllocated) {
@@ -480,8 +480,8 @@ std::optional<Config> GemmPluginProfiler<Config, RunnerPtr, GemmIdType, GemmIdHa
480480
return std::nullopt;
481481
}
482482

483-
int const n = mDims.n;
484-
int const k = mDims.k;
483+
int const n = static_cast<int>(mDims.n);
484+
int const k = static_cast<int>(mDims.k);
485485
size_t const workspace_bytes = computeTmpSize(target, n, k);
486486

487487
CUDA_CALL_THROW(cudaStreamCreate(&mStream));

0 commit comments

Comments
 (0)