Skip to content

Commit

Permalink
Add proto definitions for GPU fallbacks
Browse files Browse the repository at this point in the history
  • Loading branch information
irfansharif committed Sep 3, 2024
1 parent 89d4a00 commit 9951709
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions modal_proto/api.proto
Original file line number Diff line number Diff line change
Expand Up @@ -943,7 +943,6 @@ message FileEntry {
uint64 size = 4;
}


message Function {
string module_name = 1;
string function_name = 2;
Expand Down Expand Up @@ -1045,15 +1044,17 @@ message Function {

ClassParameterInfo class_parameter_info = 56;

repeated Resources _experimental_resources = 57; // overrides `resources` field above

reserved 57; // _experimental_resources
reserved 58;
reserved 59;
uint32 batch_max_size = 60; // Maximum number of inputs to fetch at once
uint64 batch_linger_ms = 61; // Miliseconds to block before a response is needed
bool i6pn_enabled = 62;
bool _experimental_concurrent_cancellations = 63;
uint32 max_concurrent_inputs = 64;

bool _experimental_task_templates_enabled = 65; // forces going through the new gpu-fallbacks integration path, even if no fallback options are specified
repeated TaskTemplate _experimental_task_templates = 66; // for fallback options, where the first/most-preferred "template" is derived from fields above
}

message FunctionBindParamsRequest {
Expand Down Expand Up @@ -1762,7 +1763,7 @@ message Sandbox {
// to look at fine-grained placement constraints.
reserved 16; // _experimental_scheduler
optional SchedulerPlacement scheduler_placement = 17;
repeated Resources _experimental_resources = 18; // overrides `resources` field above
reserved 18; // _experimental_resources

string worker_id = 19; // for internal debugging use only
oneof open_ports_oneof {
Expand Down Expand Up @@ -2072,6 +2073,11 @@ message TaskStats {
double started_at = 4;
}

message TaskTemplate {
uint32 priority = 1;
Resources resources = 2;
uint32 concurrent_inputs = 3;
}

message TokenFlowCreateRequest {
string utm_source = 3;
Expand Down

0 comments on commit 9951709

Please sign in to comment.