chore: add source_ir in slice layer name (#3284)

jiwoong-choi · web-flow · commit eb6d410cb34f · 2024-11-08T12:10:37.000-07:00
diff --git a/py/torch_tensorrt/dynamo/conversion/impl/slice/base.py b/py/torch_tensorrt/dynamo/conversion/impl/slice/base.py
@@ -32,5 +32,5 @@ def slice(
     )
     if dynamic_shape:
         layer.set_input(2, shape)
-    set_layer_name(layer, target, name)
+    set_layer_name(layer, target, name, source_ir)
     return layer.get_output(0)

Original file line number	Diff line number	Diff line change
`@@ -32,5 +32,5 @@ def slice(`
`32`	`32`	`)`
`33`	`33`	`if dynamic_shape:`
`34`	`34`	`layer.set_input(2, shape)`
`35`		`- set_layer_name(layer, target, name)`
	`35`	`+ set_layer_name(layer, target, name, source_ir)`
`36`	`36`	`return layer.get_output(0)`