From 9c2c06d78f7566ed412e9fecd044fb93066c3c19 Mon Sep 17 00:00:00 2001
From: Guillaume Klein <guillaume.klein@systrangroup.com>
Date: Thu, 7 Sep 2023 13:39:22 +0200
Subject: [PATCH] Fix MSVC compilation

---
 src/ops/position_encodings_add.cc | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/src/ops/position_encodings_add.cc b/src/ops/position_encodings_add.cc
index 4896f41a6..280f3f05f 100644
--- a/src/ops/position_encodings_add.cc
+++ b/src/ops/position_encodings_add.cc
@@ -30,18 +30,20 @@ namespace ctranslate2 {
 
       output.resize_as(input);
 
-      DEVICE_AND_FLOAT_DISPATCH(
-        "PositionEncodingsAdd", input.device(), input.dtype(),
-        ({
-          if (offsets)
-            compute<D, T>(step, offsets, input, encodings, output);
-          else
-            primitives<D>::add_batch_broadcast(encodings.data<T>() + step * depth,
-                                               input.data<T>(),
-                                               output.data<T>(),
-                                               time * depth,
-                                               input.size());
-        }));
+      if (offsets) {
+        DEVICE_AND_FLOAT_DISPATCH(
+          "PositionEncodingsAdd", input.device(), input.dtype(),
+          (compute<D, T>(step, offsets, input, encodings, output)));
+
+      } else {
+        DEVICE_AND_FLOAT_DISPATCH(
+          "PositionEncodingsAdd", input.device(), input.dtype(),
+          (primitives<D>::add_batch_broadcast(encodings.data<T>() + step * depth,
+                                              input.data<T>(),
+                                              output.data<T>(),
+                                              time * depth,
+                                              input.size())));
+      }
     }
 
   }