Skip to content

Commit

Permalink
Merge branch 'INSTX-5529_reduced_CUDA_batch_timeout_v6.0' into 'relea…
Browse files Browse the repository at this point in the history
…se-v0.6'

INSTX-5529 reduced cuda caller batch timeout from 60s to 30s

See merge request machine-learning/dorado!1075
  • Loading branch information
MarkBicknellONT committed Jun 18, 2024
2 parents da9fdcf + 2aac117 commit c65f1f3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dorado/basecall/CudaCaller.h
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ class CudaCaller {
size_t num_batch_dims() const { return m_batch_dims.size(); };
c10::Device device() const { return m_options.device(); }
const CRFModelConfig &config() const { return m_config; }
int batch_timeout_ms() const { return m_low_latency ? 100 : 60000; }
int batch_timeout_ms() const { return m_low_latency ? 100 : 30000; }

std::string get_name() const { return std::string("CudaCaller_") + m_device; }

Expand Down

0 comments on commit c65f1f3

Please sign in to comment.