Skip to content

Commit dde66e6

Browse files
add bool scalar type to int implicit cast (#2571)
[LTC] Add bool scalar type to int implicit cast
1 parent c61f0bd commit dde66e6

File tree

1 file changed

+1
-1
lines changed
  • projects/pt1/python/torch_mlir/csrc/base_lazy_backend/utils

1 file changed

+1
-1
lines changed

projects/pt1/python/torch_mlir/csrc/base_lazy_backend/utils/jit_utils.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ void ConvertScalarImplicit(std::shared_ptr<Graph>& graph) {
2020

2121
NodeKind node_type;
2222
TypePtr output_type;
23-
if (c10::isIntegralType(*scalar_type, false)) {
23+
if (c10::isIntegralType(*scalar_type, true)) {
2424
node_type = c10::aten::IntImplicit;
2525
output_type = IntType::get();
2626
} else if (c10::isFloatingType(*scalar_type)) {

0 commit comments

Comments
 (0)