@@ -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