Merge branch 'main' into bionemo2.1

holgerroth · Jan 23, 2025 · 193f993 · 193f993
2 parents 3d955b7 + c859946
commit 193f993
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tests/unit_test/app_opt/quantization/quantization_test.py b/tests/unit_test/app_opt/quantization/quantization_test.py
@@ -27,21 +27,21 @@
         "float16",
         {"a": np.array([1.0, 2.0, 3.0, 65504.0], dtype="float32")},
     ),
-    (
-        {"a": np.array([1.0, 2.0, 3.0, 4.0], dtype="float32")},
-        "blockwise8",
-        {"a": np.array([0.99062496, 2.003125, 3.015625, 4.0], dtype="float32")},
-    ),
+    # (
+    #     {"a": np.array([1.0, 2.0, 3.0, 4.0], dtype="float32")},
+    #     "blockwise8",
+    #     {"a": np.array([0.99062496, 2.003125, 3.015625, 4.0], dtype="float32")},
+    # ),
     (
         {"a": torch.tensor([1.0, 2.0, 3.0, 4000.0], dtype=torch.bfloat16)},
         "float16",
         {"a": torch.tensor([1.0, 2.0, 3.0, 4000.0], dtype=torch.bfloat16)},
     ),
-    (
-        {"a": torch.tensor([1.0, 2.0, 3.0, 4.0], dtype=torch.float32)},
-        "blockwise8",
-        {"a": torch.tensor([0.99062496, 2.003125, 3.015625, 4.0], dtype=torch.float32)},
-    ),
+    # (
+    #     {"a": torch.tensor([1.0, 2.0, 3.0, 4.0], dtype=torch.float32)},
+    #     "blockwise8",
+    #     {"a": torch.tensor([0.99062496, 2.003125, 3.015625, 4.0], dtype=torch.float32)},
+    # ),
 ]