From 64cda83196afabe324dff36e3b601735f49d92dd Mon Sep 17 00:00:00 2001
From: Tom Rix <trix@redhat.com>
Date: Thu, 16 May 2024 14:07:10 -0600
Subject: [PATCH] Add cuda arches to build for

Suggested-by: Steve Royer <sroyer@redhat.com>
Signed-off-by: Tom Rix <trix@redhat.com>
---
 python-torch.spec | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python-torch.spec b/python-torch.spec
index c371814..e8f451b 100644
--- a/python-torch.spec
+++ b/python-torch.spec
@@ -681,6 +681,8 @@ export BUILD_CAFFE2=ON
 export CUDACXX=/usr/local/cuda/bin/nvcc
 export CPLUS_INCLUDE_PATH=/usr/local/cuda/include
 export USE_CUDA=ON
+# The arches to build for
+export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0"
 %endif
 %endif