Skip to content

Commit

Permalink
Updates compat TF protos by running our update.sh script.
Browse files Browse the repository at this point in the history
  • Loading branch information
arcra committed Oct 19, 2023
1 parent c97601a commit ab4858f
Show file tree
Hide file tree
Showing 5 changed files with 52 additions and 5 deletions.
24 changes: 21 additions & 3 deletions tensorboard/compat/proto/config.proto
Original file line number Diff line number Diff line change
Expand Up @@ -106,8 +106,8 @@ message GPUOptions {
// Per "virtual" device memory limit, in MB. The number of elements in
// the list is the number of virtual devices to create on the
// corresponding visible GPU (see "virtual_devices" below).
// If empty, it will create single virtual device taking all available
// memory from the device.
// If empty and `num_virtual_devices_per_gpu` is not set, it will create
// single virtual device taking all available memory from the device.
//
// For the concept of "visible" and "virtual" GPU, see the comments for
// "visible_device_list" above for more information.
Expand Down Expand Up @@ -172,6 +172,12 @@ message GPUOptions {
// result in undefined behavior.
repeated VirtualDevices virtual_devices = 1;

// The number of virtual devices to create on each visible GPU. The
// available memory will be split equally among all virtual devices. If the
// field `memory_limit_mb` in `VirtualDevices` is not empty, this field will
// be ignored.
int32 num_virtual_devices_per_gpu = 15;

// If true, uses CUDA unified memory for memory allocations. If
// per_process_gpu_memory_fraction option is greater than 1.0, then unified
// memory is used regardless of the value for this field. See comments for
Expand Down Expand Up @@ -252,6 +258,13 @@ message GPUOptions {
// gpu_host_mem_limit_in_mb, because the default GPU host memory limit is
// quite high.
bool gpu_host_mem_disallow_growth = 14;

// Memory limit for gpu system. This can also be set by
// TF_DEVICE_MIN_SYS_MEMORY_IN_MB, which takes precedence over
// gpu_system_memory_size_in_mb. With this, user can configure the gpu
// system memory size for better resource estimation of multi-tenancy(one
// gpu with multiple model) use case.
int32 gpu_system_memory_size_in_mb = 16;
}

// Everything inside experimental is subject to change and is not subject
Expand Down Expand Up @@ -691,9 +704,14 @@ message ConfigProto {
// aims to negate its value.
bool disable_optimize_for_static_graph = 24;

// Whether eager remote execution will stream all the function calls or
// allow them to happen in parallel. When true, streaming execution is
// disabled, and parallel execution is allowed.
bool disable_eager_executor_streaming_enqueue = 26;

reserved 25;

// Next: 26
// Next: 27
}

Experimental experimental = 16;
Expand Down
6 changes: 6 additions & 0 deletions tensorboard/compat/proto/coordination_config.proto
Original file line number Diff line number Diff line change
Expand Up @@ -61,4 +61,10 @@ message CoordinationServiceConfig {
// silently. This is useful when we know that a task can immediately resume
// work upon re-connecting to the service.
bool allow_new_incarnation_to_reconnect = 11;

// Disables coordination service.
// Some libraries enable coordination service by default even if the user did
// not specify any config. This field allows users to explicitly disable
// coordination service under all situations.
bool force_disable = 12;
}
12 changes: 12 additions & 0 deletions tensorboard/compat/proto/types.proto
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,12 @@ enum DataType {
DT_FLOAT8_E5M2 = 24; // 5 exponent bits, 2 mantissa bits.
DT_FLOAT8_E4M3FN = 25; // 4 exponent bits, 3 mantissa bits, finite-only, with
// 2 NaNs (0bS1111111).
// TODO - b/299182407: Leaving room for remaining float8 types.
// DT_FLOAT8_E4M3FNUZ = 26;
// DT_FLOAT8_E4M3B11FNUZ = 27;
// DT_FLOAT8_E5M2FNUZ = 28;
DT_INT4 = 29;
DT_UINT4 = 30;

// Do not use! These are only for TF1's obsolete reference Variables.
// Every enum above should have a corresponding value below (verified by
Expand Down Expand Up @@ -72,6 +78,12 @@ enum DataType {
DT_UINT64_REF = 123;
DT_FLOAT8_E5M2_REF = 124;
DT_FLOAT8_E4M3FN_REF = 125;
// TODO - b/299182407: Leaving room for remaining float8 types.
// DT_FLOAT8_E4M3FNUZ_REF = 126;
// DT_FLOAT8_E4M3B11FNUZ_REF = 127;
// DT_FLOAT8_E5M2FNUZ_REF = 128;
DT_INT4_REF = 129;
DT_UINT4_REF = 130;
}
// DISABLED.ThenChange(
// https://www.tensorflow.org/code/tensorflow/c/tf_datatype.h,
Expand Down
Binary file modified tensorboard/data/server/descriptor.bin
Binary file not shown.
15 changes: 13 additions & 2 deletions tensorboard/data/server/tensorboard.pb.rs

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit ab4858f

Please sign in to comment.