fix: make compress/decompress of BaguaTensor method string consiste…

…nt (#33)
BaguaSys · Jul 13, 2021 · ee929df · ee929df
1 parent ff5e988
commit ee929df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bagua-core-internal/src/datatypes/mod.rs b/bagua-core-internal/src/datatypes/mod.rs
@@ -795,7 +795,7 @@ impl BaguaTensor {
 
     pub fn compress(&self, method: &str, n_chunks: usize, target_chunk: i32) -> Self {
         match method {
-            "min_max_uint8" => Self {
+            "MinMaxUInt8" => Self {
                 inner: Arc::new(RwLock::new(BaguaTensorInner {
                     name: "compressed_tensor".to_string(),
                     raw: self
@@ -855,7 +855,7 @@ impl BaguaTensor {
     //
     pub fn decompress_from(&mut self, method: &str, n_chunks: usize, compressed_buffer: &Self) {
         match method {
-            "min_max_uint8" => {
+            "MinMaxUInt8" => {
                 self.inner.write().raw.decompress_from(
                     &TensorCompressionMethod::MinMaxUInt8(MinMaxUInt8CompressionParameters {}),
                     n_chunks,