Skip to content

Commit 9a20ebd

Browse files
committed
Updated the virtual devices config proto.
1 parent 5c7db1e commit 9a20ebd

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

modules/proto/src/main/proto/config.proto

+12
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,18 @@ message GPUOptions {
110110
// For the concept of "visible" and "virtual" GPU, see the comments for
111111
// "visible_device_list" above for more information.
112112
repeated float memory_limit_mb = 1;
113+
114+
// Priority values to use with the virtual devices. Use the cuda function
115+
// cudaDeviceGetStreamPriorityRange to query for valid range of values for
116+
// priority.
117+
//
118+
// On a P4000 GPU with cuda 10.1, the priority range reported was 0 for
119+
// least priority and -1 for greatest priority.
120+
//
121+
// If this field is not specified, then the virtual devices will be
122+
// created with the default. If this field has values set, then the size
123+
// of this must match with the above memory_limit_mb.
124+
repeated int32 priority = 2;
113125
}
114126

115127
// The multi virtual device settings. If empty (not set), it will create

0 commit comments

Comments
 (0)