chore: Add supports_dynamic_shapes flag for where ops

keehyuna · keehyuna · commit f29596c06113 · 2024-06-07T10:30:39.000+09:00
diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -667,7 +667,7 @@ def aten_ops_split(
     )
 
 
-@dynamo_tensorrt_converter(torch.ops.aten.where.self)
+@dynamo_tensorrt_converter(torch.ops.aten.where.self, supports_dynamic_shapes=True)
 def aten_ops_where(
     ctx: ConversionContext,
     target: Target,

Original file line number	Diff line number	Diff line change
`@@ -667,7 +667,7 @@ def aten_ops_split(`
`667`	`667`	`)`
`668`	`668`
`669`	`669`
`670`		`-@dynamo_tensorrt_converter(torch.ops.aten.where.self)`
	`670`	`+@dynamo_tensorrt_converter(torch.ops.aten.where.self, supports_dynamic_shapes=True)`
`671`	`671`	`def aten_ops_where(`
`672`	`672`	`ctx: ConversionContext,`
`673`	`673`	`target: Target,`