lint

microsoft · Nov 7, 2023 · 4a96881 · 4a96881
1 parent 4b3bec1
commit 4a96881
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/contrib_ops/cuda/bert/group_query_attention.h b/onnxruntime/contrib_ops/cuda/bert/group_query_attention.h
@@ -20,8 +20,8 @@ class GroupQueryAttention final : public CudaKernel {
   Status ComputeInternal(OpKernelContext* context) const override;
 
  protected:
-  int num_heads_;           // number of attention heads
-  int kv_num_heads_;        // different for k and v for group query attention
+  int num_heads_;     // number of attention heads
+  int kv_num_heads_;  // different for k and v for group query attention
   // bool left_padding_;       // shifts last token to end of buffer
   bool is_unidirectional_;  // causal
   bool is_past_bsnh_;