patch out FLT_MAX

recipe/patches/0001-point-to-correct-prefix.patch

@@ -1,7 +1,7 @@
 From 8aa5c7343a493513cb95e09d10bab17849787aaf Mon Sep 17 00:00:00 2001
 From: Hadrien Mary <hadrien.mary@gmail.com>
 Date: Fri, 27 Sep 2024 23:15:12 -0400
-Subject: [PATCH 1/4] point to correct prefix
+Subject: [PATCH 1/5] point to correct prefix
 
 ---
  tools/setup_helpers/extension.py | 5 +++--

recipe/patches/0002-use-conda-cuda.patch

@@ -1,7 +1,7 @@
 From a1ee9df1cfa2d47eff3d74da2f588e3a4befc6eb Mon Sep 17 00:00:00 2001
 From: Tobias Fischer <info@tobiasfischer.info>
 Date: Thu, 10 Oct 2024 07:48:28 +1000
-Subject: [PATCH 2/4] use conda cuda
+Subject: [PATCH 2/5] use conda cuda
 
 ---
  tools/setup_helpers/extension.py | 4 ----

recipe/patches/0003-Apply-CMAKE_ARGS-if-set.patch

@@ -1,7 +1,7 @@
 From c7ced84f1175294d75f55ae84a761f9d29720378 Mon Sep 17 00:00:00 2001
 From: Tobias Fischer <info@tobiasfischer.info>
 Date: Thu, 10 Oct 2024 09:38:57 +1000
-Subject: [PATCH 3/4] Apply CMAKE_ARGS if set
+Subject: [PATCH 3/5] Apply CMAKE_ARGS if set
 
 ---
  tools/setup_helpers/extension.py | 4 ++++

recipe/patches/0004-Add-missing-skipIfNoFFmpeg-for-TestFileObject.patch

@@ -1,7 +1,7 @@
 From 9fc1da385a19aaeda50b80d6a0023832f885a494 Mon Sep 17 00:00:00 2001
 From: Tobias Fischer <info@tobiasfischer.info>
 Date: Sat, 5 Oct 2024 20:52:42 +1000
-Subject: [PATCH 4/4] Add missing @skipIfNoFFmpeg for TestFileObject
+Subject: [PATCH 4/5] Add missing @skipIfNoFFmpeg for TestFileObject
 
 ---
  test/torchaudio_unittest/backend/dispatcher/ffmpeg/info_test.py | 1 +

recipe/patches/0005-replace-FLT_MAX-for-compatibility-with-newer-cudatoo.patch

@@ -0,0 +1,85 @@
+From 737c672b87157ad9d644f8150c8685bc294a7c85 Mon Sep 17 00:00:00 2001
+From: "H. Vetinari" <h.vetinari@gmx.com>
+Date: Sun, 24 Nov 2024 17:57:45 +1100
+Subject: [PATCH 5/5] replace FLT_MAX for compatibility with newer cudatoolkit
+
+---
+ .../cuctc/src/ctc_prefix_decoder_kernel_v2.cu  | 18 +++++++++---------
+ 1 file changed, 9 insertions(+), 9 deletions(-)
+
+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 4ca8f1bf..0a9d0fb4 100644
+--- a/src/libtorchaudio/cuctc/src/ctc_prefix_decoder_kernel_v2.cu
++++ b/src/libtorchaudio/cuctc/src/ctc_prefix_decoder_kernel_v2.cu
+@@ -151,7 +151,7 @@ __global__ void prob_matrix_v2_kernel(
+       } else {
+         out_prob = _logprob(cur_prob, _logsumexp(beamid_p.x, beamid_p.y));
+       }
+-      ptable[idout] = -FLT_MAX;
++      ptable[idout] = -std::numeric_limits<float>::max();
+       ptablen[idout] = out_prob;
+     }
+   }
+@@ -185,7 +185,7 @@ __global__ void prob_space_blank_kernel_v2(
+     int idout = blid + (batch_id * beam + beamid) * ldc;
+     ptable[idout] = _logprob(pc, _logsumexp(tmpprev.x, tmpprev.y));
+     if (last_char == blid)
+-      ptablen[idout] = -FLT_MAX;
++      ptablen[idout] = -std::numeric_limits<float>::max();
+   }
+
+   if (spid >= 0 && (spid != blid) && beamid < beam) {
+@@ -194,7 +194,7 @@ __global__ void prob_space_blank_kernel_v2(
+     int idout = spid + (batch_id * beam + beamid) * ldc;
+     ptablen[idout] = _lauguage() *
+         _logprob(pc, _logsumexp(tmpprev.x, tmpprev.y)); // logsumexp
+-    ptable[idout] = -FLT_MAX;
++    ptable[idout] = -std::numeric_limits<float>::max();
+   }
+ }
+
+@@ -246,8 +246,8 @@ __global__ void matrix_merge_kernel_v2(
+
+       ptable[tidout] = _logsumexp(ptable[tidout], ptable[tidin]);
+       ptablen[tidout] = _logsumexp(ptablen[tidout], ptablen[tidin]);
+-      ptable[tidin] = -FLT_MAX;
+-      ptablen[tidin] = -FLT_MAX;
++      ptable[tidin] = -std::numeric_limits<float>::max();
++      ptablen[tidin] = -std::numeric_limits<float>::max();
+     }
+   }
+ }
+@@ -304,13 +304,13 @@ __global__ __launch_bounds__(BLOCK_SIZE) void first_matrix__bitonic_topk_kernel(
+     int shift = clen[idx + batch_id * ldbeam];
+     if (id != blid) {
+       float2 xy =
+-          is_need_add_blank ? float2{key, -FLT_MAX} : float2{-FLT_MAX, key};
++          is_need_add_blank ? float2{key, -std::numeric_limits<float>::max()} : float2{-std::numeric_limits<float>::max(), key};
+       pprev[batch_id * ldbeam + idx] = xy;
+       clist[batch_id * beam * ldseq_len + idx * ldseq_len + shift] = id;
+       clen[batch_id * ldbeam + idx] += 1;
+       clast[batch_id * ldbeam + idx] = id;
+     } else {
+-      pprev[batch_id * ldbeam + idx] = float2{key, -FLT_MAX};
++      pprev[batch_id * ldbeam + idx] = float2{key, -std::numeric_limits<float>::max()};
+     }
+     score[batch_id * ldbeam + idx] = key;
+   }
+@@ -493,7 +493,7 @@ __launch_bounds__(BLOCK_SIZE) void topk_reduce_and_copy_list_per_batch_kernel(
+       ptable_ptablen.y = ptablen[batch_id * ldc * beam + id];
+       float cur_score = _logsumexp(ptable_ptablen.x, ptable_ptablen.y);
+       score[batch_id * ldbeam + out_beamid] = cur_score;
+-      float2 ptable_ptablen2 = float2{cur_score, -FLT_MAX};
++      float2 ptable_ptablen2 = float2{cur_score, -std::numeric_limits<float>::max()};
+       pprev[batch_id * ldbeam + out_beamid] =
+           is_need_add_blank ? ptable_ptablen2 : ptable_ptablen;
+     }
+@@ -528,7 +528,7 @@ __launch_bounds__(BLOCK_SIZE) void topk_reduce_and_copy_list_per_batch_kernel(
+         ptable_ptablen.y = ptablen[batch_id * ldc * beam + id];
+         float cur_score = _logsumexp(ptable_ptablen.x, ptable_ptablen.y);
+         score[batch_id * ldbeam + idx] = cur_score;
+-        float2 ptable_ptablen2 = float2{cur_score, -FLT_MAX};
++        float2 ptable_ptablen2 = float2{cur_score, -std::numeric_limits<float>::max()};
+         pprev[batch_id * ldbeam + idx] =
+             is_need_add_blank ? ptable_ptablen2 : ptable_ptablen;
+       }

recipe/recipe.yaml

@@ -21,6 +21,7 @@ source:
       - patches/0002-use-conda-cuda.patch
       - patches/0003-Apply-CMAKE_ARGS-if-set.patch
       - patches/0004-Add-missing-skipIfNoFFmpeg-for-TestFileObject.patch
+      - patches/0005-replace-FLT_MAX-for-compatibility-with-newer-cudatoo.patch
 
 build:
   number: ${{ build_number }}