diff --git a/xla/python/ifrt/executable.h b/xla/python/ifrt/executable.h index 879305ea4aee8..e544ebef3e3d0 100644 --- a/xla/python/ifrt/executable.h +++ b/xla/python/ifrt/executable.h @@ -125,7 +125,7 @@ struct ExecuteOptions { // Execution stream ID identifies the series of executions that must be // executed in program order. Executions with different execution stream IDs // may be executed in any order and concurrently. - uint64_t execution_stream_id = 0; + int64_t execution_stream_id = 0; // Custom execution options specific to the runtime. The user and the runtime // are responsible for ensuring version compatibility. diff --git a/xla/python/ifrt/execute_options.proto b/xla/python/ifrt/execute_options.proto index 93a98e00b5c4c..832b19b61f0eb 100644 --- a/xla/python/ifrt/execute_options.proto +++ b/xla/python/ifrt/execute_options.proto @@ -24,7 +24,7 @@ message ExecuteOptionsProto { int32 launch_id = 3; repeated int32 non_donatable_input_indices = 7; bool fill_status = 9; - uint64 execution_stream_id = 11; + int64 execution_stream_id = 11; AttributeMapProto custom_options = 10; reserved 1, 4 to 6, 8;