mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
Fix the test and add custom min/max reductions for uncommon MPI types (#2060)
This commit is contained in:
committed by
GitHub
parent
dfae2c6989
commit
ddaa4b7dcb
@@ -53,7 +53,8 @@ void AllReduce::eval_cpu(
|
||||
distributed::detail::all_min(group(), in, outputs[0], stream());
|
||||
break;
|
||||
default:
|
||||
throw std::runtime_error("Only all reduce sum, min and max are supported for now");
|
||||
throw std::runtime_error(
|
||||
"Only all reduce sum, min and max are supported for now");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user