biondizzle
c2b752c2fe
- nvfp4_mega_moe_full: drop-in replacement for deep_gemm.mega.fp8_nvfp4_mega_moe - transform_nvfp4_weights_for_mega_moe: weight transformation (tested) - SymmBuffer + get_symm_buffer_for_nvfp4_mega_moe: API-matching stubs - MEGA_MOE_STATIC=1 support for pipeline testing - pyproject.toml for pip install
17 lines
371 B
TOML
17 lines
371 B
TOML
[build-system]
|
|
requires = ["setuptools>=68.0", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "nvfp4-megamoe-kernel"
|
|
version = "0.1.0"
|
|
description = "NVFP4 Mega MoE kernel for DeepSeek-V4-Pro on Blackwell (TileLang)"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
"torch>=2.5",
|
|
"tilelang>=0.1",
|
|
]
|
|
|
|
[tool.setuptools.packages.find]
|
|
where = ["src"]
|