File size: 331 Bytes
25e8424 | 1 2 3 4 5 6 7 | [pytest]
testpaths = tests
markers =
sim: NKI CPU simulator (nki.simulate); needs nki, numpy, ml_dtypes. Not device evidence.
cpu: CPU-only check against the Hugging Face model; needs torch + transformers (downloads the model).
neuron: runs on a Trainium2 device under PyTorch Native (torch.compile(backend="neuron")).
|