We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
get_ir
1 parent 80bbd8b commit 18dcdd0Copy full SHA for 18dcdd0
py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -32,7 +32,7 @@ def get_ir(target: Target) -> SourceIR:
32
target_module = getattr(target, "__module__", "None")
33
if any(
34
target_module.startswith(prefix)
35
- for prefix in ("torch.ops.prims", "torch._ops.prims")
+ for prefix in ("torch.ops.aten", "torch._ops.aten")
36
):
37
return SourceIR.ATEN
38
elif any(
0 commit comments