mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-04-20 06:18:59 +00:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
"""AMD/ROCm backend tests for KT-Kernel (Placeholder).
|
|
|
|
This file is a placeholder for future AMD/ROCm backend tests.
|
|
Currently, KT-Kernel focuses on CPU optimizations (Intel AMX/AVX512).
|
|
|
|
To implement AMD tests:
|
|
1. Add actual test functions with @pytest.mark.amd
|
|
2. Update the estimated time in register_amd_ci()
|
|
3. Implement AMD/ROCm-specific initialization and validation tests
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
# Add parent directory to path for CI registration
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
|
from ci.ci_register import register_amd_ci
|
|
|
|
# Register this test for AMD CI (estimated time: 10 seconds, placeholder)
|
|
# Update suite name when implementing: currently using "stage-a-test-1"
|
|
register_amd_ci(est_time=10, suite="stage-a-test-1")
|
|
|
|
|
|
def test_amd_placeholder():
|
|
"""Placeholder test for AMD/ROCm backend.
|
|
|
|
TODO: Implement actual AMD/ROCm tests when AMD support is added to kt-kernel.
|
|
"""
|
|
# Currently a no-op placeholder
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Allow running standalone (required by test runner)
|
|
print("⚠ AMD/ROCm tests are not yet implemented (placeholder)")
|
|
print("✓ Placeholder test passed")
|