From f441831a35a79eb676b087e3beb2784cf2032efc Mon Sep 17 00:00:00 2001
From: Shehroze Khan <shehroze.khan@cerebras.net>
Date: Mon, 13 Nov 2023 11:18:04 -0800
Subject: [PATCH] add bool scalar type to int implicit cast

---
 .../torch_mlir/csrc/base_lazy_backend/utils/jit_utils.cpp       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/pt1/python/torch_mlir/csrc/base_lazy_backend/utils/jit_utils.cpp b/projects/pt1/python/torch_mlir/csrc/base_lazy_backend/utils/jit_utils.cpp
index 8d64f9fb7c9b..9ca8b666a42e 100644
--- a/projects/pt1/python/torch_mlir/csrc/base_lazy_backend/utils/jit_utils.cpp
+++ b/projects/pt1/python/torch_mlir/csrc/base_lazy_backend/utils/jit_utils.cpp
@@ -20,7 +20,7 @@ void ConvertScalarImplicit(std::shared_ptr<Graph>& graph) {
 
     NodeKind node_type;
     TypePtr output_type;
-    if (c10::isIntegralType(*scalar_type, false)) {
+    if (c10::isIntegralType(*scalar_type, true)) {
       node_type = c10::aten::IntImplicit;
       output_type = IntType::get();
     } else if (c10::isFloatingType(*scalar_type)) {