File tree 1 file changed +12
-0
lines changed
modules/proto/src/main/proto
1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -110,6 +110,18 @@ message GPUOptions {
110
110
// For the concept of "visible" and "virtual" GPU, see the comments for
111
111
// "visible_device_list" above for more information.
112
112
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 ;
113
125
}
114
126
115
127
// The multi virtual device settings. If empty (not set), it will create
You can’t perform that action at this time.
0 commit comments