[ZenCPU] AMD Zen CPU Backend with supported dtypes via zentorch weekly (#39967)

Signed-off-by: Chinmay Kulkarni <Chinmay.Kulkarni@amd.com>
This commit is contained in:
Chinmay-Kulkarni-AMD
2026-04-18 11:52:37 +05:30
committed by GitHub
parent aeee7ef939
commit 87518c3027
2 changed files with 11 additions and 1 deletions
+3 -1
View File
@@ -1085,7 +1085,9 @@ setup(
install_requires=get_requirements(),
extras_require={
# AMD Zen CPU optimizations via zentorch
"zen": ["zentorch"],
"zen": [
"zentorch-weekly==5.2.1.dev20260408"
], # Zentorch has weekly releases. This pulls the known-good version.
"bench": ["pandas", "matplotlib", "seaborn", "datasets", "scipy", "plotly"],
"tensorizer": ["tensorizer==2.10.1"],
"fastsafetensors": ["fastsafetensors >= 0.2.2"],