Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions crates/tower-cmd/src/run.rs
Original file line number Diff line number Diff line change
Expand Up @@ -322,9 +322,12 @@ fn build_cli_execution_spec(
parameters: params,
env_vars,
resources: ResourceLimits {
cpu_millicores: None,
memory_mb: None,
storage_mb: None,
cpu_limit_millicores: None,
cpu_request_millicores: None,
memory_limit_mb: None,
memory_request_mb: None,
storage_limit_mb: None,
storage_request_mb: None,
max_pids: None,
gpu_count: 0,
timeout_seconds: 3600,
Expand Down
18 changes: 14 additions & 4 deletions crates/tower-runtime/src/execution.rs
Original file line number Diff line number Diff line change
Expand Up @@ -114,17 +114,27 @@ pub enum CacheBackend {
None,
}

/// ResourceLimits defines compute resource constraints
/// ResourceLimits defines compute resource requests and limits.
/// Missing values are resolved by the execution backend.
#[derive(Debug, Clone)]
pub struct ResourceLimits {
/// CPU limit in millicores (e.g., 1000 = 1 CPU)
pub cpu_millicores: Option<u32>,
pub cpu_limit_millicores: Option<u32>,

/// CPU request in millicores
pub cpu_request_millicores: Option<u32>,

Comment thread
konstantinoscs marked this conversation as resolved.
Resolved
/// Memory limit in megabytes
pub memory_mb: Option<u32>,
pub memory_limit_mb: Option<u32>,

/// Memory request in megabytes
pub memory_request_mb: Option<u32>,

/// Ephemeral storage limit in megabytes
pub storage_mb: Option<u32>,
pub storage_limit_mb: Option<u32>,

/// Ephemeral storage request in megabytes
pub storage_request_mb: Option<u32>,
Comment thread
konstantinoscs marked this conversation as resolved.
Resolved

/// Maximum number of processes
pub max_pids: Option<u32>,
Expand Down
9 changes: 6 additions & 3 deletions crates/tower-runtime/tests/subprocess_test.rs
Original file line number Diff line number Diff line change
Expand Up @@ -65,9 +65,12 @@ async fn create_execution_spec(id: String, package: Package) -> ExecutionSpec {
command: None,
},
resources: ResourceLimits {
cpu_millicores: None,
memory_mb: None,
storage_mb: None,
cpu_limit_millicores: None,
cpu_request_millicores: None,
memory_limit_mb: None,
memory_request_mb: None,
storage_limit_mb: None,
storage_request_mb: None,
max_pids: None,
gpu_count: 0,
timeout_seconds: 300,
Expand Down
Loading