Skip to content

Commit ab4858f

Browse files
committed
Updates compat TF protos by running our update.sh script.
1 parent c97601a commit ab4858f

File tree

5 files changed

+52
-5
lines changed

5 files changed

+52
-5
lines changed

tensorboard/compat/proto/config.proto

Lines changed: 21 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,8 @@ message GPUOptions {
106106
// Per "virtual" device memory limit, in MB. The number of elements in
107107
// the list is the number of virtual devices to create on the
108108
// corresponding visible GPU (see "virtual_devices" below).
109-
// If empty, it will create single virtual device taking all available
110-
// memory from the device.
109+
// If empty and `num_virtual_devices_per_gpu` is not set, it will create
110+
// single virtual device taking all available memory from the device.
111111
//
112112
// For the concept of "visible" and "virtual" GPU, see the comments for
113113
// "visible_device_list" above for more information.
@@ -172,6 +172,12 @@ message GPUOptions {
172172
// result in undefined behavior.
173173
repeated VirtualDevices virtual_devices = 1;
174174

175+
// The number of virtual devices to create on each visible GPU. The
176+
// available memory will be split equally among all virtual devices. If the
177+
// field `memory_limit_mb` in `VirtualDevices` is not empty, this field will
178+
// be ignored.
179+
int32 num_virtual_devices_per_gpu = 15;
180+
175181
// If true, uses CUDA unified memory for memory allocations. If
176182
// per_process_gpu_memory_fraction option is greater than 1.0, then unified
177183
// memory is used regardless of the value for this field. See comments for
@@ -252,6 +258,13 @@ message GPUOptions {
252258
// gpu_host_mem_limit_in_mb, because the default GPU host memory limit is
253259
// quite high.
254260
bool gpu_host_mem_disallow_growth = 14;
261+
262+
// Memory limit for gpu system. This can also be set by
263+
// TF_DEVICE_MIN_SYS_MEMORY_IN_MB, which takes precedence over
264+
// gpu_system_memory_size_in_mb. With this, user can configure the gpu
265+
// system memory size for better resource estimation of multi-tenancy(one
266+
// gpu with multiple model) use case.
267+
int32 gpu_system_memory_size_in_mb = 16;
255268
}
256269

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

707+
// Whether eager remote execution will stream all the function calls or
708+
// allow them to happen in parallel. When true, streaming execution is
709+
// disabled, and parallel execution is allowed.
710+
bool disable_eager_executor_streaming_enqueue = 26;
711+
694712
reserved 25;
695713

696-
// Next: 26
714+
// Next: 27
697715
}
698716

699717
Experimental experimental = 16;

tensorboard/compat/proto/coordination_config.proto

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,4 +61,10 @@ message CoordinationServiceConfig {
6161
// silently. This is useful when we know that a task can immediately resume
6262
// work upon re-connecting to the service.
6363
bool allow_new_incarnation_to_reconnect = 11;
64+
65+
// Disables coordination service.
66+
// Some libraries enable coordination service by default even if the user did
67+
// not specify any config. This field allows users to explicitly disable
68+
// coordination service under all situations.
69+
bool force_disable = 12;
6470
}

tensorboard/compat/proto/types.proto

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,12 @@ enum DataType {
4343
DT_FLOAT8_E5M2 = 24; // 5 exponent bits, 2 mantissa bits.
4444
DT_FLOAT8_E4M3FN = 25; // 4 exponent bits, 3 mantissa bits, finite-only, with
4545
// 2 NaNs (0bS1111111).
46+
// TODO - b/299182407: Leaving room for remaining float8 types.
47+
// DT_FLOAT8_E4M3FNUZ = 26;
48+
// DT_FLOAT8_E4M3B11FNUZ = 27;
49+
// DT_FLOAT8_E5M2FNUZ = 28;
50+
DT_INT4 = 29;
51+
DT_UINT4 = 30;
4652

4753
// Do not use! These are only for TF1's obsolete reference Variables.
4854
// Every enum above should have a corresponding value below (verified by
@@ -72,6 +78,12 @@ enum DataType {
7278
DT_UINT64_REF = 123;
7379
DT_FLOAT8_E5M2_REF = 124;
7480
DT_FLOAT8_E4M3FN_REF = 125;
81+
// TODO - b/299182407: Leaving room for remaining float8 types.
82+
// DT_FLOAT8_E4M3FNUZ_REF = 126;
83+
// DT_FLOAT8_E4M3B11FNUZ_REF = 127;
84+
// DT_FLOAT8_E5M2FNUZ_REF = 128;
85+
DT_INT4_REF = 129;
86+
DT_UINT4_REF = 130;
7587
}
7688
// DISABLED.ThenChange(
7789
// https://www.tensorflow.org/code/tensorflow/c/tf_datatype.h,
535 Bytes
Binary file not shown.

tensorboard/data/server/tensorboard.pb.rs

Lines changed: 13 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)