From 3bd54db700d705783c129a9a729d5289c141d41a Mon Sep 17 00:00:00 2001 From: Aleksandr Borzunov Date: Fri, 30 Jun 2023 01:01:53 +0000 Subject: [PATCH] Skip BLOCKWISE_8BIT in lossy_sequential test until https://github.com/learning-at-home/hivemind/issues/572 is fixed --- tests/test_remote_sequential.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_remote_sequential.py b/tests/test_remote_sequential.py index 734683fb6..3c8a48f58 100644 --- 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