Skip to content

Commit 23ce522

Browse files
committed
fix: add fmod
1 parent dc21249 commit 23ce522

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

src/05computation/src/operators/simple_binary.cc

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,10 @@ namespace refactor::computation {
4343
static uint8_t ID = 9;
4444
return reinterpret_cast<size_t>(&ID);
4545
}
46+
case Ty::Fmod: {
47+
static uint8_t ID = 10;
48+
return reinterpret_cast<size_t>(&ID);
49+
}
4650
default:
4751
UNREACHABLE();
4852
}
@@ -70,6 +74,8 @@ namespace refactor::computation {
7074
return "Xor";
7175
case Ty::Mod:
7276
return "Mod";
77+
case Ty::Fmod:
78+
return "Fmod";
7379
default:
7480
UNREACHABLE();
7581
}

src/07onnx/src/operators/simple_binary.cc

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,10 @@ namespace refactor::onnx {
6565
static uint8_t ID = 9;
6666
return reinterpret_cast<size_t>(&ID);
6767
}
68+
case Ty::Fmod: {
69+
static uint8_t ID = 10;
70+
return reinterpret_cast<size_t>(&ID);
71+
}
6872
default:
6973
UNREACHABLE();
7074
}
@@ -83,6 +87,7 @@ namespace refactor::onnx {
8387
case Ty::Or : return "onnx::Or" ;
8488
case Ty::Xor: return "onnx::Xor";
8589
case Ty::Mod: return "onnx::Mod";
90+
case Ty::Fmod: return "onnx::Mod";
8691
default: UNREACHABLE();
8792
}
8893
// clang-format on
@@ -181,6 +186,7 @@ namespace refactor::onnx {
181186
case Ty::Or : type_ = Ty_::Or ; break;
182187
case Ty::Xor : type_ = Ty_::Xor; break;
183188
case Ty::Mod : type_ = Ty_::Mod; break;
189+
case Ty::Fmod : type_ = Ty_::Fmod; break;
184190
default: UNREACHABLE();
185191
}
186192
// clang-format on

0 commit comments

Comments
 (0)