some linting

torchtitan/experiments/deepseek_v3/generate.py

@@ -20,9 +20,9 @@
 from model_config import deepseek_config_registry
 from torch.distributed.device_mesh import DeviceMesh
 from torch.distributed.pipelining import PipelineStage, ScheduleGPipe
+from transformers import AutoTokenizer
 
 from torchtitan.tools.utils import Color
-from transformers import AutoTokenizer
 
 # Uncomment the model you want to run.
 model_id, mesh_shape = "deepseek-ai/DeepSeek-V2-Lite-Chat", (1, 4)

torchtitan/experiments/kernels/blackwell/cute_grouped_gemm_fwd.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
 """
     Strategy using CUTLASS GroupedGemmKernel for group GEMM operations on Blackwell.
 
@@ -467,10 +473,12 @@ def _execute_cutlass_kernel(
         num_groups = len(problem_sizes)
 
         # Convert to CUTE tensors using improved converter
-        problem_sizes_cute, strides_cute, ptrs_cute = (
-            self.converter.create_metadata_tensors(
-                problem_sizes, strides_abc, ptrs_abc, device
-            )
+        (
+            problem_sizes_cute,
+            strides_cute,
+            ptrs_cute,
+        ) = self.converter.create_metadata_tensors(
+            problem_sizes, strides_abc, ptrs_abc, device
         )
 
         # Get other required components

torchtitan/experiments/kernels/blackwell/group_gemm_base.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
 import torch
 
 

torchtitan/experiments/kernels/blackwell/pytorch_cute_converter.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
 import logging
 from typing import Any, Dict, List, Optional, Tuple