diff --git a/csrc/preseg_passes/translate_repeat_to_expand.cpp b/csrc/preseg_passes/translate_repeat_to_expand.cpp index 71b0e0b061a..382dcb85f52 100644 --- a/csrc/preseg_passes/translate_repeat_to_expand.cpp +++ b/csrc/preseg_passes/translate_repeat_to_expand.cpp @@ -160,7 +160,7 @@ class RepeatToExpandTranslator { std::find(inp_domain.begin(), inp_domain.end(), info.repeated_id)); bcast_flags.at(repeated_id_offset) = true; auto broadcast_tv = broadcast(info.input_tv, bcast_flags); - NVF_ERROR(broadcast_tv->nDims() == inp_domain.size() + 1); + NVF_ERROR((size_t)broadcast_tv->nDims() == inp_domain.size() + 1); // Step 2 std::vector expanded_sizes(