mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-01 08:48:46 +02:00
ef4c5eb65e
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
59 lines
1.6 KiB
Protocol Buffer
59 lines
1.6 KiB
Protocol Buffer
syntax = "proto3";
|
|
|
|
package tensorflow;
|
|
|
|
option cc_enable_arenas = true;
|
|
option java_outer_classname = "DeviceAttributesProtos";
|
|
option java_multiple_files = true;
|
|
option java_package = "org.tensorflow.framework";
|
|
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/device_attributes_go_proto";
|
|
|
|
message InterconnectLink {
|
|
int32 device_id = 1;
|
|
string type = 2;
|
|
int32 strength = 3;
|
|
}
|
|
|
|
message LocalLinks {
|
|
repeated InterconnectLink link = 1;
|
|
}
|
|
|
|
message DeviceLocality {
|
|
// Optional bus locality of device. Default value of 0 means
|
|
// no specific locality. Specific localities are indexed from 1.
|
|
int32 bus_id = 1;
|
|
|
|
// Optional NUMA locality of device.
|
|
int32 numa_node = 2;
|
|
|
|
// Optional local interconnect links to other devices.
|
|
LocalLinks links = 3;
|
|
}
|
|
|
|
message DeviceAttributes {
|
|
// Fully specified name of the device within a cluster.
|
|
string name = 1;
|
|
|
|
// String representation of device_type.
|
|
string device_type = 2;
|
|
|
|
// Memory capacity of device in bytes.
|
|
int64 memory_limit = 4;
|
|
|
|
// Platform-specific data about device that may be useful
|
|
// for supporting efficient data transfers.
|
|
DeviceLocality locality = 5;
|
|
|
|
// A device is assigned a global unique number each time it is
|
|
// initialized. "incarnation" should never be 0.
|
|
fixed64 incarnation = 6;
|
|
|
|
// String representation of the physical device that this device maps to.
|
|
string physical_device_desc = 7;
|
|
|
|
// A physical device ID for use in XLA DeviceAssignments, unique across
|
|
// clients in a multi-client setup. Set to -1 if unavailable, non-negative
|
|
// otherwise.
|
|
int64 xla_global_id = 8;
|
|
}
|