radixark/miles

Largest clones in the code

• 119 lines in 2 placespython817 tokens
  miles_plugins/models/deepseek_v4/ops/kernel/tilelang_indexer_fwd.py:7–125·miles_plugins/models/glm5/ops/tilelang_indexer_fwd.py:3–121

  7import tilelang
  8import torch
  9from tilelang import language as T
  10
  11
  12@tilelang.jit(
  13    pass_configs={
  14        tilelang.PassConfigKey.TL_ENABLE_FAST_MATH: True,
  15    },
  16)
  

  First 10 of 119 lines.

• 74 lines in 2 placespython553 tokens
  miles_plugins/models/deepseek_v4/ops/kernel/tilelang_indexer_bwd.py:3–76·miles_plugins/models/glm5/ops/tilelang_indexer_bwd.py:3–76

  3import tilelang as tl
  4import tilelang.language as T
  5import torch
  6
  7BF16 = T.bfloat16
  8FP32 = T.float32
  9INT32 = T.int32
  10
  11pass_configs = {
  12    tl.PassConfigKey.TL_DISABLE_TMA_LOWER: True,
  

  First 10 of 74 lines.

• 48 lines in 3 placespython268 tokens
  miles/backends/megatron_utils/megatron_to_hf/deepseekv3.py:6–53·miles/backends/megatron_utils/megatron_to_hf/qwen3_next.py:6–53·miles/backends/megatron_utils/megatron_to_hf/qwen3moe.py:6–53

  6def convert_deepseekv3_to_hf(args, name, param):
  7    if name == "module.module.embedding.word_embeddings.weight":
  8        return [("model.embed_tokens.weight", param)]
  9    if name == "module.module.output_layer.weight":
  10        return [("lm_head.weight", param)]
  11    if name == "module.module.decoder.final_layernorm.weight":
  12        return [("model.norm.weight", param)]
  13
  14    try:
  15        head_dim = args.kv_channels if args.kv_channels is not None else args.hidden_size // args.num_attention_heads
  

  First 10 of 48 lines.