mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-29 07:17:08 +08:00
760e3cadc0
- tests/neuron/setup_trn2.sh: one-shot setup of a fresh Trn2 (venv layered
on the DLAMI torch-neuronx venv, C++ build, official reference + tokenizer,
tiny checkpoint, CPU golden).
- tests/neuron/run_all.sh cpu|device|all [stage...]: vendor kernel tests,
tilelang_ops interface tests (basic-mode stubs reported as skips via
nie_skip), deep_ep at world size 1/2/4 (gloo) and 2/4/8 on the neuron
process group, device primitive checks, end-to-end vs golden on CPU and
on a NeuronCore, and the device diff; PASS/FAIL summary.
- tests/neuron/device/test_device_primitives.py: torch-neuronx behaviours
the port depends on or works around (fp8 strided-view indexing, uint64,
1-rank all_to_all, inference-mode version counters, ...).
- examples/neuron/device_diff.py: re-runs every kernel call of a NeuronCore
run on host copies and ranks kernels by device/host difference.
- Unit tests written during bring-up moved from scratch dirs into
tests/neuron/{deep_gemm,flash_mla,deep_ep,tile_kernels,sampling}.
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
16 lines
800 B
Python
16 lines
800 B
Python
import os, sys
|
|
sys.path.insert(0, os.path.join(os.path.normpath(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "..", "..", "..")), "deepinfer", "neuron_ops", "vendor"))
|
|
_D = os.path.dirname(os.path.abspath(__file__))
|
|
os.environ.setdefault("TK_TEST_MERGE_MODE", "overwrite")
|
|
os.environ.setdefault("TK_BENCHMARK_TESTS_DIR", _D)
|
|
os.environ.setdefault("TK_BENCHMARK_BASELINES_DIR", os.path.join(_D, "bb"))
|
|
os.environ.setdefault("TK_BENCHMARK_FAIL_MODE", "none")
|
|
os.environ.setdefault("TK_GPU_MEM_PROFILES_DIR", os.path.join(_D, "gm"))
|
|
pytest_plugins = [
|
|
"tile_kernels.testing.pytest.benchmark",
|
|
"tile_kernels.testing.pytest.gpu_mem",
|
|
"tile_kernels.testing.pytest.random",
|
|
"tile_kernels.testing.pytest.tilelang_warning",
|
|
"tile_kernels.testing.pytest.xdist_failfast",
|
|
]
|