Unverified Commit 2350968e authored by Crutcher Dunnavant's avatar Crutcher Dunnavant Committed by GitHub
Browse files

Move f/utils => f/internal; move testing libs to fair_dev/testing (#1004)

parent 3b727945
...@@ -12,9 +12,9 @@ from unittest import mock ...@@ -12,9 +12,9 @@ from unittest import mock
from parameterized import parameterized from parameterized import parameterized
import torch import torch
from fairscale.utils import torch_version from fair_dev.testing.testing import dist_init, spawn_for_all_world_sizes
from fairscale.utils.reduce_scatter_bucketer import ReduceScatterBucketer from fairscale.internal import torch_version
from fairscale.utils.testing import dist_init, spawn_for_all_world_sizes from fairscale.internal.reduce_scatter_bucketer import ReduceScatterBucketer
def rename_test(testcase_func, param_num, param): def rename_test(testcase_func, param_num, param):
......
...@@ -12,7 +12,7 @@ ...@@ -12,7 +12,7 @@
import torch import torch
from torch import nn from torch import nn
from fairscale.utils.state_dict import find_module_instances, replace_by_prefix_ from fairscale.internal.state_dict import find_module_instances, replace_by_prefix_
def test_find_module_instances(): def test_find_module_instances():
......
from fairscale.utils import torch_version from fairscale.internal import torch_version
def test_torch_version(): def test_torch_version():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment