From b1481ba21bc01f6492b22435b67b03a2ddc3c346 Mon Sep 17 00:00:00 2001 From: Joel Lamy-Poirier Date: Thu, 4 Jun 2026 15:34:49 -0400 Subject: [PATCH 1/2] Re-enable dropless MoE test The dropless MoE kernel was fixed in #489 but the dedicated test_dropless_mlp skip was left behind. Remove it; the test passes on GPU. Co-Authored-By: Claude Opus 4.8 (1M context) --- tests/functional/test_functional.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/functional/test_functional.py b/tests/functional/test_functional.py index 07b8768da..bd0414d11 100644 --- a/tests/functional/test_functional.py +++ b/tests/functional/test_functional.py @@ -87,7 +87,6 @@ def test_mlp_recomputation(gated, activation, testing_device): # Takes ~6s, much more if it needs to compile, reducing the hidden size doesn't help. @pytest.mark.slow -@pytest.mark.skip("Dropless MoE is broken") def test_dropless_mlp(testing_device): device = torch.device(testing_device) num_experts = 4 From 698e7f05378d7980f0813f895f9df9708bb69feb Mon Sep 17 00:00:00 2001 From: Joel Lamy-Poirier Date: Thu, 4 Jun 2026 15:34:52 -0400 Subject: [PATCH 2/2] Fix VISION extra install on aarch64 pillow-simd is an x86-only (SSE4) fork with no aarch64 build, so the VISION extra fails to install on ARM. Gate it to x86_64 via an environment marker and fall back to stock Pillow on other architectures. Co-Authored-By: Claude Opus 4.8 (1M context) --- setup.cfg | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/setup.cfg b/setup.cfg index bab3eebff..2a51a439f 100644 --- a/setup.cfg +++ b/setup.cfg @@ -68,7 +68,9 @@ STREAMING = VISION = # Vision Tools webp>=0.4.0 - pillow-simd>=9.5.0 + # pillow-simd is an x86-only (SSE4) fork; fall back to stock Pillow on other architectures (e.g. aarch64). + pillow-simd>=9.5.0; platform_machine == "x86_64" + Pillow>=9.5.0; platform_machine != "x86_64" torchvision>=0.24.0 DEV =