-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
sci-libs/torchaudio: Fix CUDA 12.5 build
See pytorch/audio#3811 Signed-off-by: wangjiezhe <wangjiezhe@gmail.com>
- Loading branch information
1 parent
9d0a0a6
commit 498f67b
Showing
2 changed files
with
29 additions
and
1 deletion.
There are no files selected for viewing
24 changes: 24 additions & 0 deletions
24
sci-libs/torchaudio/files/torchaudio-2.5.0-Fix-CUDA-12.5-build.patch
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,24 @@ | ||
From 7797f83e1d66ff78872763e1da3a5fb2f0534c40 Mon Sep 17 00:00:00 2001 | ||
From: Markus Hennerbichler <markushennerbichler@gmail.com> | ||
Date: Mon, 15 Jul 2024 14:07:13 +0100 | ||
Subject: [PATCH] Fix CUDA 12.5 build | ||
|
||
CUDA 12.5 removed the FLT_MAX symbol. | ||
This was previously used without being explicitly imported. | ||
FLT_MAX is defined in <float.h>, including this header fixes the issue | ||
--- | ||
src/libtorchaudio/cuctc/src/ctc_prefix_decoder_kernel_v2.cu | 1 + | ||
1 file changed, 1 insertion(+) | ||
|
||
diff --git a/src/libtorchaudio/cuctc/src/ctc_prefix_decoder_kernel_v2.cu b/src/libtorchaudio/cuctc/src/ctc_prefix_decoder_kernel_v2.cu | ||
index 4ca8f1bf24..e6192155a2 100644 | ||
--- a/src/libtorchaudio/cuctc/src/ctc_prefix_decoder_kernel_v2.cu | ||
+++ b/src/libtorchaudio/cuctc/src/ctc_prefix_decoder_kernel_v2.cu | ||
@@ -24,6 +24,7 @@ | ||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | ||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||
#include <algorithm> | ||
+#include <float.h> | ||
#include "ctc_fast_divmod.cuh" | ||
#include "cub/cub.cuh" | ||
#include "device_data_wrap.h" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters