Add missing PROFILE block in Rotary op

OpenNMT · Sep 14, 2023 · d240717 · d240717
1 parent 98a36e8
commit d240717
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ops/rotary.cc b/src/ops/rotary.cc
@@ -15,6 +15,8 @@ namespace ctranslate2 {
                             const StorageView& sin,
                             const StorageView& cos,
                             StorageView& output) const {
+      PROFILE("Rotary");
+
       output.resize_as(input);
 
       DEVICE_AND_FLOAT_DISPATCH("Rotary", input.device(), input.dtype(),