Skip to content

Commit 500e246

Browse files
committed
feat: implement theming system and LLM gallery wizard with enhanced component architecture and process management utilities
1 parent c2bcb75 commit 500e246

67 files changed

Lines changed: 9941 additions & 329 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

.services.pid

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
backend=30112
2+
frontend=30113

backend.log

Lines changed: 7295 additions & 0 deletions
Large diffs are not rendered by default.

cppmega_mlx/kernels/compute_dacs_segsum.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def compute_dacs_segsum(dt: mx.array, A: mx.array, *,
3232
prod = dt32 * A32 # (B, T, H)
3333
# Build chunked cumulative sum: each chunk restarts.
3434
n_chunks = (T + chunk_size - 1) // chunk_size
35-
out = mx.zeros((B, T, H), dtype=mx.float32)
35+
mx.zeros((B, T, H), dtype=mx.float32)
3636
out_list = []
3737
for c in range(n_chunks):
3838
start = c * chunk_size

cppmega_mlx/nn/_tilelang/dsa_splitk_indexer_loss.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1583,7 +1583,6 @@ def _bench_stage2_q_hoist_wave5(
15831583
reason}`` when prerequisites are missing.
15841584
"""
15851585

1586-
import time
15871586

15881587
ok, msg = _tilelang_available()
15891588
if not ok:

cppmega_mlx/nn/_tilelang/fp8_amax.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,6 @@
5757
from __future__ import annotations
5858

5959
import threading
60-
import types
6160
from dataclasses import dataclass
6261
from functools import lru_cache
6362
from typing import Any, cast

cppmega_mlx/nn/_tilelang/sparse_mla_path_c.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1487,7 +1487,7 @@ def _canonicalize_bwd_path_b_hot_loops(
14871487
) -> str:
14881488
"""Make backward hot loops match the hand-written Path B MSL shape."""
14891489

1490-
threads = _threadgroup_size(topk)
1490+
_threadgroup_size(topk)
14911491
kv_base = "uint kv_row_base = kv_b_base + (uint(gather_idx) * kv_group + g) * qk_dim;"
14921492
msl = msl.replace(
14931493
(

cppmega_mlx/nn/dsa_indexer_loss_reference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def reference_indexer_kl_loss(
5656
raise ValueError("Q and K must share batch dim")
5757
if Q.shape[2] != K.shape[2] or Q.shape[3] != K.shape[3]:
5858
raise ValueError("Q and K must share (AH, AD)")
59-
B, Sq, AH, AD = (int(Q.shape[0]), int(Q.shape[1]),
59+
_B, Sq, _AH, _AD = (int(Q.shape[0]), int(Q.shape[1]),
6060
int(Q.shape[2]), int(Q.shape[3]))
6161
Sk = int(K.shape[1])
6262

cppmega_mlx/runtime/path_c_fusion_launcher.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from __future__ import annotations
2828

2929
import json
30-
from collections.abc import Mapping, Sequence
30+
from collections.abc import Mapping
3131
from dataclasses import dataclass
3232
from typing import Any
3333

cppmega_mlx/training/distributed_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838

3939
from __future__ import annotations
4040

41-
from typing import Any, Callable
41+
from typing import Any
4242

4343
import mlx.core as mx
4444
import mlx.nn as nn

cppmega_mlx/training/plasticity/redo.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,7 @@
1717

1818
from __future__ import annotations
1919

20-
from collections.abc import Callable, Iterable
21-
from typing import Any
20+
from collections.abc import Callable
2221

2322
import mlx.core as mx
2423
import mlx.nn as nn

0 commit comments

Comments
 (0)