diff --git a/examples/eager_mlp/mlp_eager_simple.py b/examples/eager_mlp/mlp_eager_simple.py index d5bbb3443..947d0da6e 100644 --- a/examples/eager_mlp/mlp_eager_simple.py +++ b/examples/eager_mlp/mlp_eager_simple.py @@ -13,7 +13,7 @@ import torchvision.transforms as transforms import torchvision.datasets as datasets -torch._dynamo.config.dynamic_shapes = False +torch._dynamo.config.dynamic_shapes = False # TODO: https://github.com/nod-ai/SHARK-Turbine/issues/93 class MNISTDataLoader: diff --git a/tests/dynamo/mninst_test.py b/tests/dynamo/mninst_test.py index 6c599ed61..7374dde13 100644 --- a/tests/dynamo/mninst_test.py +++ b/tests/dynamo/mninst_test.py @@ -13,7 +13,10 @@ import torchvision.transforms as transforms import torchvision.datasets as datasets -torch._dynamo.config.dynamic_shapes = False + +import torch._dynamo.config + +torch._dynamo.config.dynamic_shapes = False # TODO: https://github.com/nod-ai/SHARK-Turbine/issues/93 class MNISTDataLoader: