[torchlib] Improve aten::fill

microsoft · Jul 25, 2024 · be86593 · be86593
1 parent 712aa87
commit be86593
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -3622,10 +3622,6 @@ def aten_full(
 
     if dtype != -1:
         fill_value = op.Cast(fill_value, to=dtype)
-    if isinstance(size, list) and size == []:
-        # TODO(justinchuby): Handle empty list better than using isinstance
-        # size can be empty, meaning a scalar
-        return fill_value
 
     size = op.Cast(size, to=INT64.dtype)
     return op.Expand(fill_value, size)