hotfix: fix typo (#56)

jasperzhong · Jul 30, 2020 · ced6f2b · ced6f2b
1 parent 8b1218d
commit ced6f2b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/byteps/common/compressor/impl/randomk.cc b/byteps/common/compressor/impl/randomk.cc
@@ -29,7 +29,7 @@ CompressorRegistry::Register reg(
       auto factor = HyperParamFinder<float>(kwargs, "compressor_k", false,
                                             [](float x) { return x > 0; });
       unsigned k;
-      if (k < 1) {
+      if (factor < 1) {
         k = static_cast<unsigned>(factor * size / getDataTypeLength(dtype));
       } else {
         k = static_cast<unsigned>(factor);

diff --git a/byteps/common/compressor/impl/topk.cc b/byteps/common/compressor/impl/topk.cc
@@ -30,7 +30,7 @@ CompressorRegistry::Register reg(
       auto factor = HyperParamFinder<float>(kwargs, "compressor_k", false,
                                             [](float x) { return x > 0; });
       unsigned k;
-      if (k < 1) {
+      if (factor < 1) {
         k = static_cast<unsigned>(factor * size / getDataTypeLength(dtype));
       } else {
         k = static_cast<unsigned>(factor);