Skip BLOCKWISE_8BIT in lossy_sequential test until

learning-at-home/hivemind#572 is fixed
bigscience-workshop · Jun 30, 2023 · 3bd54db · 3bd54db
1 parent f8b32ec
commit 3bd54db
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_remote_sequential.py b/tests/test_remote_sequential.py
@@ -78,7 +78,10 @@ def get_request_metadata(self, protocol: str, *args, **kwargs):
         if protocol == "rpc_forward":
             metadata["output_compression"] = (runtime_pb2.CompressionType.FLOAT16,)
         elif protocol == "rpc_backward":
-            metadata["output_compression"] = (runtime_pb2.CompressionType.BLOCKWISE_8BIT,)
+            metadata["output_compression"] = (runtime_pb2.CompressionType.FLOAT16,)
+            # FIXME: Initially, we used CompressionType.BLOCKWISE_8BIT for rpc_backward() here.
+            # This is currently broken since hivemind==1.1.8 is not compatible with bitsandbytes==0.39.1.
+            # Please revert to BLOCKWISE_8BIT once this is fixed: https://github.com/learning-at-home/hivemind/issues/572
         return metadata