lint

NVIDIA · Dec 18, 2024 · c4ca266 · c4ca266
1 parent 38721fe
commit c4ca266
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/csrc/host_ir/executor.cpp b/csrc/host_ir/executor.cpp
@@ -274,8 +274,9 @@ void HostIrEvaluator::handle(SetCurrentStream* set_current_stream) {
 }
 
 void HostIrEvaluator::handle(GetCurrentStream* get_current_stream) {
-  c10::DeviceIndex my_device_index =
-      communicator_ ? communicator_->deviceId() : 0;
+  auto my_device_index = communicator_
+      ? static_cast<c10::DeviceIndex>(communicator_->deviceId())
+      : 0;
   streams_.insert(
       {get_current_stream->stream(),
        c10::cuda::getCurrentCUDAStream(my_device_index)});