File tree 1 file changed +20
-1
lines changed
1 file changed +20
-1
lines changed Original file line number Diff line number Diff line change @@ -673,6 +673,25 @@ message ConfigProto {
673
673
// Whether runtime execution uses TFRT.
674
674
bool use_tfrt = 18 ;
675
675
676
+ // If true, use Pathways with TFRT API for multi host support.
677
+ bool enable_multi_host = 27 ;
678
+
679
+ // Port for the Pathways server. Ignored if enable_multi_host=false.
680
+ int32 backend_server_port = 28 ;
681
+
682
+ // If true, TFRT will use TPU specific compiler passes and perform TPU
683
+ // specific initialization.
684
+ bool target_tpu = 29 ;
685
+
686
+ // If true, TFRT will use GPU specific compiler passes and perform GPU
687
+ // specific initialization.
688
+ bool target_gpu = 30 ;
689
+
690
+ // The threshold to merge small streams in TFRT. The stream with cost
691
+ // smaller than the threshold will be merged. Setting it to value 1
692
+ // disables all merges.
693
+ int32 stream_merge_threshold = 31 ;
694
+
676
695
// The field "coordination_service was previously specified as a string;
677
696
// this has been replaced with a message below.
678
697
reserved 19 ;
@@ -711,7 +730,7 @@ message ConfigProto {
711
730
712
731
reserved 25 ;
713
732
714
- // Next: 27
733
+ // Next: 32
715
734
}
716
735
717
736
Experimental experimental = 16 ;
You can’t perform that action at this time.
0 commit comments