p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

support scalar bool

This commit is contained in:
huangbingjian 2023-01-18 09:34:19 +08:00
parent 931030e2fa
commit 57a89773da
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/frontend/operator/cc_implementations.cc
View File

@ -216,6 +216,10 @@ bool InnerScalarGe(T x, U y) {
int64_t sum = InnerScalar##op_t(IntToLong(GetValue<int>(x)), GetValue<int64_t>(y)); \
return MakeValue(sum); \
} \
if (x->isa<BoolImm>() && y->isa<BoolImm>()) { \
int sum = InnerScalar##op_t(static_cast<int>(GetValue<bool>(x)), static_cast<int>(GetValue<bool>(y))); \
return MakeValue(sum); \
} \
MS_EXCEPTION(TypeError) << "Unsupported input type for Scalar" << #op_t << ", type of x:" << x->type_name() \
<< ", value of x:" << x->ToString() << ", type of y:" << y->type_name() \
<< ", value of y:" << y->ToString(); \