From bcdd1822e39ef62207489ac40554d02f21115c3c Mon Sep 17 00:00:00 2001
From: Tom Rix <trix@redhat.com>
Date: Tue, 25 Jun 2024 08:09:08 -0400
Subject: [PATCH] Add CUDA BuildRequires

Signed-off-by: Tom Rix <trix@redhat.com>
---
 python-torch.spec | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/python-torch.spec b/python-torch.spec
index 39e5c44..2c0e093 100644
--- a/python-torch.spec
+++ b/python-torch.spec
@@ -100,7 +100,12 @@
 # Pick a version that works
 %if %{with cuda}
 %if 0%{?fedora} < 40
-%global cuda_ver 12.4
+%global cuda_ver     12.5
+%global cudart_ver   12-5
+%global cublas_ver   12-5
+%global cufft_ver    12-5
+%global curand_ver   12-5
+%global cusparse_ver 12-5
 %else
 %global cuda_ver 12.5
 %endif
@@ -378,6 +383,14 @@ BuildRequires:  roctracer-devel
 Requires:       rocm-rpm-macros-modules
 %endif
 
+%if %{with cuda}
+BuildRequires:  cuda-cudart-devel-%{cudart_ver}
+BuildRequires:  libcublas-devel-%{cublas_ver}
+BuildRequires:  libcufft-devel-%{cufft_ver}
+BuildRequires:  libcurand-devel-%{curand_ver}
+BuildRequires:  libcusparse-devel-%{cusparse_ver}
+%endif
+
 %if %{with opencv}
 BuildRequires:  opencv-devel
 %endif