first commit

flash_qla/utils/__init__.py

@@ -0,0 +1,20 @@
+# Copyright (c) 2026 The Qwen team, Alibaba Group.
+# Licensed under The MIT License [see LICENSE for details]
+
+from .profiler import profile
+from .pack import pad_and_reshape, pack, unpack, fill_last_chunk_of_g
+from .math import l2norm
+from .index import prepare_chunk_indices, prepare_chunk_offsets, tensor_cache
+
+
+__all__ = [
+    "profile",
+    "pad_and_reshape",
+    "pack",
+    "unpack",
+    "fill_last_chunk_of_g",
+    "l2norm",
+    "prepare_chunk_indices",
+    "prepare_chunk_offsets",
+    "tensor_cache",
+]