Incremented version

nict-wisdom · Jan 26, 2022 · 3176fca · 3176fca
1 parent 3cd7e70
commit 3176fca
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/setup.py b/setup.py
@@ -67,7 +67,7 @@ def build_extension(self, ext):
         subprocess.check_call(['cmake', '--build', '.'] + build_args, cwd=self.build_temp)
 
 
-VERSION = "0.7.2"
+VERSION = "0.7.3"
 
 if "CUDA_VERSION" in os.environ:
     version_nums = os.environ["CUDA_VERSION"].split(".")

diff --git a/test/test_simple.py b/test/test_simple.py
@@ -14,14 +14,14 @@
 
 test_models = [
     {"model": models.SmallParamModel},
-    # {"model": models.SharedInputModel},
-    # {"model": models.ForkJoinModel},
-    # {"model": models.SharedParamModel},
-    # {"model": models.OneOpModel},
-    # {"model": models.TensorMulModel},
-    # {"model": models.EmbeddingModel, "get_dataset": models.EmbeddingModel.get_dataset},
-    # {"model": models.FunctionModel, "get_dataset": models.FunctionModel.get_dataset},
-    # {"model": models.LossOutModel, "loss_out": True},
+    {"model": models.SharedInputModel},
+    {"model": models.ForkJoinModel},
+    {"model": models.SharedParamModel},
+    {"model": models.OneOpModel},
+    {"model": models.TensorMulModel},
+    {"model": models.EmbeddingModel, "get_dataset": models.EmbeddingModel.get_dataset},
+    {"model": models.FunctionModel, "get_dataset": models.FunctionModel.get_dataset},
+    {"model": models.LossOutModel, "loss_out": True},
     # {"model": models.BasicModel},
     # {"model": native_models.NativeCallModel01}, # compiles module
     # {"model": models.LayerNormModel, "preprocess": models.norm_to_float} # DP only
@@ -30,8 +30,8 @@
 
 @pytest.mark.parametrize("test_model", test_models)
 @pytest.mark.parametrize("gradient_accumulation_steps", [1])
-@pytest.mark.parametrize("use_amp", [False])
-@pytest.mark.parametrize("allreduce_amp_master_params", [False])
+@pytest.mark.parametrize("use_amp", [False, True])
+@pytest.mark.parametrize("allreduce_amp_master_params", [False, True])
 @pytest.mark.parametrize("enable_zero", [False])
 @pytest.mark.parametrize("dist_params", [False])
 @pytest.mark.parametrize("offload_params", [False])