Fix incorrect innode being checked (#1273)

Co-authored-by: Gabriele Oliaro <goliaro@cs.cmu.edu>
flexflow · Jan 20, 2024 · 57d1883 · 57d1883
1 parent 75edadc
commit 57d1883
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/flexflow/torch/model.py b/python/flexflow/torch/model.py
@@ -955,7 +955,7 @@ def is_left_scalar_op(node):
         if len(innodes) != 2:
             return False
         return type(innodes[0]) is float or \
-            type(innodes[1]) is int
+            type(innodes[0]) is int
 
     @staticmethod
     def is_elemwise_op(node):