tlopex opened a new pull request, #19819:
URL: https://github.com/apache/tvm/pull/19819
tvm.testing's test-gating layer had a number of one-line helper functions
that add a name but no behavior. Inline the thin ones so call sites name the
underlying flag/feature/condition directly.
Pytest plugin (plugin.py): _target_to_requirement built its skip / gpu marks
through two one-line wrappers (_gpu_mark_and_skip / _skip_only) plus a per-kind
if ladder. Replace them with two frozensets (GPU- vs CPU-family kinds) and
resolve the skip probe by name:
marks.append(pytest.mark.skipif(not getattr(env, f"has_{kind}")(),
reason=f"need {kind}"))
The cuda+cudnn / cuda+cublas accelerator-library cases are remapped inline
(cudnn before cublas). Adds two direct unit tests for the cudnn/cublas
special-case and the unknown-kind ([]) fallback.
tvm.testing.env (env.py): inline the pure probe wrappers that just forwarded
to a primitive --
* build-flag (5): has_cutlass/rpc/nnapi/openclml/mrvl ->
env.build_flag_enabled ("USE_X"). The private _build_flag_enabled is promoted
to the public build_flag_enabled; the composed probes
(has_cudnn/cublas/nccl/hipblas) and the hexagon/adreno probes call it too.
* cpu-feature (5 pure): has_arm_dot/arm_fp16/aarch64_sve/aarch64_sme/x86_amx
->
env.has_cpu_feature("..."). The composed has_x86_vnni (avx512vnni OR
avxvnni) and has_x86_avx512 (a five-feature set) are kept -- not thin wrappers.
Also drops the obsolete test_build_flag_probe_matches_libinfo self-test and
the matching _BOOL_PROBES entries.
The runtime device probes (has_cuda/has_rocm/...) are intentionally left
as-is: the pytest plugin resolves env.has_<kind>() from each target kind, so
those names are load-bearing rather than thin wrappers.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
