fix compilation

OpenNMT · Feb 28, 2024 · 76ebbfc · 76ebbfc
1 parent 5e9dc28
commit 76ebbfc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/layers/attention.cc b/src/layers/attention.cc
@@ -601,7 +601,8 @@ namespace ctranslate2 {
       if (_tensor_parallel) {
         Shape shape = output.shape();
         StorageView tmp(std::move(shape), output.dtype(), output.device());
-        ops::ReduceAll(ops::ReduceAll::RED_OP::SUM)(output, tmp);
+        ops::ReduceAll ops_reduce_all(ops::ReduceAll::RED_OP::SUM);
+        ops_reduce_all(output, tmp);
         output = std::move(tmp);
       }
       if (_layer_norm) {