monster/trellis-2-mrp-mlx
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

test: isolate standalone Metal integration smoke
Some checks are pending
CodeQL Advanced / Analyze (${{ matrix.language }}) (none, c-cpp) (push) Waiting to run
Details
CodeQL Advanced / Analyze (${{ matrix.language }}) (none, python) (push) Waiting to run
Details

Browse Source
This commit is contained in:
Jourloy 2026-07-17 10:09:01 +03:00
parent 847bd59c25
commit 754d403e62
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pytest.ini
View File

@ -1,3 +0,0 @@
[pytest]
testpaths = tests
python_files = test_*.py

8
test_flex_gemm_integration.py
View File

@ -16,6 +16,14 @@ torch.zeros_like which lacks an MPS kernel in some PyTorch builds. That is a
PyTorch issue, separate from the mtlgemm fix being verified here. PyTorch issue, separate from the mtlgemm fix being verified here.
""" """
# This is an intentionally standalone, verbose MPS integration program rather
# than a pytest module. Avoid executing its device setup during pytest
# collection; CI and macOS acceptance invoke it directly.
if __name__ != "__main__":
import pytest
pytest.skip("run test_flex_gemm_integration.py directly", allow_module_level=True)
import os import os
os.environ.setdefault("SPARSE_CONV_BACKEND", "flex_gemm") os.environ.setdefault("SPARSE_CONV_BACKEND", "flex_gemm")
os.environ.setdefault("SPARSE_ATTN_BACKEND", "sdpa") os.environ.setdefault("SPARSE_ATTN_BACKEND", "sdpa")