Skip to content

Commit 97daf75

Browse files
pianpwkfacebook-github-bot
authored andcommitted
refactor _Dim into Dim
Summary: forward fix T218515233 Differential Revision: D71769231
1 parent 37036b3 commit 97daf75

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/xnnpack/test/tester/tester.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -496,14 +496,14 @@ def generate_random_inputs(self):
496496
# the function to compute the derived dim.
497497
dim_spec = dim_spec.root
498498
fn = dim_spec.fn
499-
elif isinstance(dim_spec, torch.export.dynamic_shapes._Dim):
499+
elif isinstance(dim_spec, torch.export.dynamic_shapes.Dim):
500500
# Not derived dim so fn is just itself
501501
def fn(x):
502502
return x
503503

504504
else:
505505
raise RuntimeError(
506-
f"Expected Dynamic Dims to be of type _DerivedDim or _Dim but got {type(dim_spec)}"
506+
f"Expected Dynamic Dims to be of type _DerivedDim or Dim but got {type(dim_spec)}"
507507
)
508508
dim_name = dim_spec.__name__
509509
if dim_name not in dim_name_to_size:

0 commit comments

Comments
 (0)