Restrict single-threading to single sequence

gdalle · Nov 15, 2024 · 4c126d8 · 4c126d8
1 parent 67343ac
commit 4c126d8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/inference/forward.jl b/src/inference/forward.jl
@@ -123,7 +123,7 @@ function forward!(
     control_seq::AbstractVector;
     seq_ends::AbstractVectorOrNTuple{Int},
 )
-    if seq_ends isa NTuple
+    if seq_ends isa NTuple{1}
         for k in eachindex(seq_ends)
             _forward!(storage, hmm, obs_seq, control_seq, seq_ends, k)
         end

diff --git a/src/inference/forward_backward.jl b/src/inference/forward_backward.jl
@@ -81,7 +81,7 @@ function forward_backward!(
     seq_ends::AbstractVectorOrNTuple{Int},
     transition_marginals::Bool=true,
 )
-    if seq_ends isa NTuple
+    if seq_ends isa NTuple{1}
         for k in eachindex(seq_ends)
             _forward_backward!(
                 storage, hmm, obs_seq, control_seq, seq_ends, k; transition_marginals

diff --git a/src/inference/viterbi.jl b/src/inference/viterbi.jl
@@ -83,7 +83,7 @@ function viterbi!(
     control_seq::AbstractVector;
     seq_ends::AbstractVectorOrNTuple{Int},
 ) where {R}
-    if seq_ends isa NTuple
+    if seq_ends isa NTuple{1}
         for k in eachindex(seq_ends)
             _viterbi!(storage, hmm, obs_seq, control_seq, seq_ends, k)
         end