mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-06 12:20:11 +01:00
[Upkeep] Resolve 4 instances of the following issue: Todo (resolved)
PiperOrigin-RevId: 821824006
This commit is contained in:
parent
03819ab09c
commit
a40f3bdebd
|
|
@ -31,7 +31,6 @@ limitations under the License.
|
||||||
|
|
||||||
namespace xla {
|
namespace xla {
|
||||||
|
|
||||||
// TODO(b/181653482): Fix for interprocedural collectives as well.
|
|
||||||
absl::StatusOr<bool> CollectivesScheduleLinearizer::Run(
|
absl::StatusOr<bool> CollectivesScheduleLinearizer::Run(
|
||||||
HloModule* module,
|
HloModule* module,
|
||||||
const absl::flat_hash_set<absl::string_view>& execution_threads) {
|
const absl::flat_hash_set<absl::string_view>& execution_threads) {
|
||||||
|
|
|
||||||
|
|
@ -1274,7 +1274,6 @@ TEST_P(CollectivePipelineParallelismTest,
|
||||||
|
|
||||||
// This is the partially pipelined version of
|
// This is the partially pipelined version of
|
||||||
// NaiveBFSMicrobatch5CircularRepeat2Replica4 and should yield the same results.
|
// NaiveBFSMicrobatch5CircularRepeat2Replica4 and should yield the same results.
|
||||||
// TODO(b/383868854): replace this with GPU pipeliner implementation.
|
|
||||||
TEST_P(CollectivePipelineParallelismTest,
|
TEST_P(CollectivePipelineParallelismTest,
|
||||||
NaiveBFSMb5Cr2Replica4SendRecvPartiallyPipelined) {
|
NaiveBFSMb5Cr2Replica4SendRecvPartiallyPipelined) {
|
||||||
constexpr char kMoreComputationsStr[] = R"(
|
constexpr char kMoreComputationsStr[] = R"(
|
||||||
|
|
@ -1508,7 +1507,6 @@ TEST_P(CollectivePipelineParallelismTest,
|
||||||
|
|
||||||
// This is the async-grouped version of
|
// This is the async-grouped version of
|
||||||
// NaiveBFSMicrobatch5CircularRepeat2Replica4 and should yield the same results.
|
// NaiveBFSMicrobatch5CircularRepeat2Replica4 and should yield the same results.
|
||||||
// TODO(b/383868854): replace this with GPU pipeliner implementation.
|
|
||||||
TEST_P(CollectivePipelineParallelismTest,
|
TEST_P(CollectivePipelineParallelismTest,
|
||||||
NaiveBFSMb5Cr2Replica4SendRecvAsyncGroup) {
|
NaiveBFSMb5Cr2Replica4SendRecvAsyncGroup) {
|
||||||
constexpr char kMoreComputationsStr[] = R"(
|
constexpr char kMoreComputationsStr[] = R"(
|
||||||
|
|
|
||||||
|
|
@ -158,8 +158,6 @@ std::string DefaultCollectiveDevicesIfEmpty(
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
// TODO(b/390097558): Add an option to generate perf table for collective which
|
|
||||||
// gets overlap to model resource contention.
|
|
||||||
int main(int argc, char* argv[]) {
|
int main(int argc, char* argv[]) {
|
||||||
// Default args.
|
// Default args.
|
||||||
int32_t num_nodes = 1;
|
int32_t num_nodes = 1;
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user