Skip to content

Commit 18dcdd0

Browse files
authored
minor fix: Update get_ir prefixes (#2369)
1 parent 80bbd8b commit 18dcdd0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def get_ir(target: Target) -> SourceIR:
3232
target_module = getattr(target, "__module__", "None")
3333
if any(
3434
target_module.startswith(prefix)
35-
for prefix in ("torch.ops.prims", "torch._ops.prims")
35+
for prefix in ("torch.ops.aten", "torch._ops.aten")
3636
):
3737
return SourceIR.ATEN
3838
elif any(

0 commit comments

Comments
 (0)