diff --git a/frontends/pytorch/test/graph_import/test_script_add3.py b/frontends/pytorch/test/graph_import/add3.py similarity index 100% rename from frontends/pytorch/test/graph_import/test_script_add3.py rename to frontends/pytorch/test/graph_import/add3.py diff --git a/frontends/pytorch/test/graph_import/test_script_debug_info.py b/frontends/pytorch/test/graph_import/debug-info.py similarity index 88% rename from frontends/pytorch/test/graph_import/test_script_debug_info.py rename to frontends/pytorch/test/graph_import/debug-info.py index 391d5be92..f343d7092 100644 --- a/frontends/pytorch/test/graph_import/test_script_debug_info.py +++ b/frontends/pytorch/test/graph_import/debug-info.py @@ -19,9 +19,9 @@ mb = torch_mlir.ModuleBuilder() def add3(t0, t1, t2): # TODO: Checks for debug info are quite hard with the new trailing debug # attribute print. See if this can be improved. - # CHECK: loc({{.*}}test_script_debug_info.py":[[# @LINE + 1]] + # CHECK: loc({{.*}}debug-info.py":[[# @LINE + 1]] intermediate = t0 + t1 - # CHECK: loc({{.*}}test_script_debug_info.py":[[# @LINE + 1]] + # CHECK: loc({{.*}}debug-info.py":[[# @LINE + 1]] final = intermediate + t2 return final diff --git a/frontends/pytorch/test/graph_import/test_errors.py b/frontends/pytorch/test/graph_import/errors.py similarity index 100% rename from frontends/pytorch/test/graph_import/test_errors.py rename to frontends/pytorch/test/graph_import/errors.py