Skip to content

[gluon][examples] MoE bmm1 in Gluon #10047

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Mogball merged 8 commits into from
Apr 17, 2026
Merged

[gluon][examples] MoE bmm1 in Gluon #10047

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
6fac6eb
bmm1 gluon example
Mogball Apr 15, 2026
94e65ac
fmt
Mogball Apr 15, 2026
3632637
nonuniform logits
Mogball Apr 16, 2026
6ae003c
remove inval
Mogball Apr 16, 2026
41a00b1
triton_kernels
Mogball Apr 16, 2026
3d0982d
flag
Mogball Apr 16, 2026
4301e71
Merge remote-tracking branch 'origin/main' into jeffniu/bmm1
Mogball Apr 17, 2026
1bce51b
fix tl_core
Mogball Apr 17, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion Makefile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ TRITON_OPT := $(BUILD_DIR)/bin/triton-opt
PYTEST := $(PYTHON) -m pytest
LLVM_BUILD_PATH ?= "$(ROOT_DIR)/.llvm-project/build"
NUM_PROCS ?= 8
TRITON_KERNELS_PATH := $(ROOT_DIR)/python/triton_kernels$(if $(PYTHONPATH),:$(PYTHONPATH))

# Incremental builds

Expand Down Expand Up @@ -50,7 +51,7 @@ test-unit: all
.PHONY: test-gluon
test-gluon: all
$(PYTEST) -n $(NUM_PROCS) python/test/gluon/ python/tutorials/gluon/
$(PYTEST) -n 2 python/examples/gluon/
PYTHONPATH="$(TRITON_KERNELS_PATH)" $(PYTEST) -n 2 python/examples/gluon/

.PHONY: test-gsan
test-gsan: all
Expand Down
Loading
Loading