Fix right trim

ml-explore · Dec 21, 2024 · 63e7267 · 63e7267
1 parent 1906cbb
commit 63e7267
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/mlx/backend/common/compiled_cpu.cpp b/mlx/backend/common/compiled_cpu.cpp
@@ -45,7 +45,13 @@ std::string exec(std::string cmd) {
     ret += buffer;
   }
   // Trim trailing spaces.
-  ret.erase(std::remove_if(ret.begin(), ret.end(), isspace), ret.end());
+  ret.erase(
+      std::find_if(
+          ret.rbegin(),
+          ret.rend(),
+          [](unsigned char ch) { return !std::isspace(ch); })
+          .base(),
+      ret.end());
   return ret;
 }