Skip to content

Commit 530e451

Browse files
committed
Arm backend: Add support for floor_divide.default
Signed-off-by: Agrima Khare <[email protected]> Change-Id: I85153b1b245862a107e6469600e69c059477f417
1 parent 1a9ab9d commit 530e451

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/arm/_passes/arm_pass_manager.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,7 @@ def _tosa_FP_pipeline(
246246
self.add_pass(DecomposeDivTensorModePass())
247247
self.add_pass(ReplaceScalarWithTensorByProfilePass())
248248
self.add_pass(DecomposeRemainderPass())
249+
self.add_pass(DecomposeFloorDividePass())
249250
self.add_pass(DecomposeDivTensorModePass())
250251
self.add_pass(DecomposeEmbeddingPass())
251252
self.add_pass(FuseQuantizedActivationPass())
@@ -339,6 +340,7 @@ def transform_for_annotation_pipeline(self, graph_module: GraphModule):
339340
self.add_pass(DecomposeFloorDividePass())
340341
self.add_pass(ReplaceScalarWithTensorByProfilePass())
341342
self.add_pass(DecomposeRemainderPass())
343+
self.add_pass(DecomposeFloorDividePass())
342344
self.add_pass(DecomposeDivTensorModePass())
343345
self.add_pass(DecomposeAddSubAlphaPass())
344346
self.add_pass(ScalarsToAttributePass())

0 commit comments

Comments
 (0)