llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add bool scalar type to int implicit cast (#2571) 

[LTC] Add bool scalar type to int implicit cast
pull/2577/head
Shehroze Khan 2023-11-14 08:56:12 -05:00 committed by GitHub
parent c61f0bd5bb
commit dde66e66b0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
projects/pt1/python/torch_mlir/csrc/base_lazy_backend/utils/jit_utils.cpp
View File

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