chore: Add supports_dynamic_shapes flag for where ops

keehyuna · keehyuna · commit 7827fa453b9f · 2024-05-30T07:16:26.000Z
diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -656,7 +656,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
`@@ -656,7 +656,7 @@ def aten_ops_split(`
`656`	`656`	`)`
`657`	`657`
`658`	`658`
`659`		`-@dynamo_tensorrt_converter(torch.ops.aten.where.self)`
	`659`	`+@dynamo_tensorrt_converter(torch.ops.aten.where.self, supports_dynamic_shapes=True)`
`660`	`660`	`def aten_ops_where(`
`661`	`661`	`ctx: ConversionContext,`
`662`	`662`	`target: Target,`