Searched refs:BROADCAST_COLLECTIVE (Results 1 – 12 of 12) sorted by relevance
25 NcclBroadcaster() : NcclBase(BROADCAST_COLLECTIVE, "NcclBroadcast") {} in NcclBroadcaster()
41 case BROADCAST_COLLECTIVE: in InitializeCollectiveParams()
323 col_params_->instance.type = BROADCAST_COLLECTIVE; in CollectiveBcastSendOpKernel()400 col_params_->instance.type = BROADCAST_COLLECTIVE; in CollectiveBcastRecvOpKernel()790 col_params->instance.type = BROADCAST_COLLECTIVE; in ComputeAsync()924 col_params->instance.type = BROADCAST_COLLECTIVE; in ComputeAsync()
484 : NcclTestBase(/*collective_type=*/BROADCAST_COLLECTIVE, in NcclBroadcasterTest()
72 if (col_params.instance.type == BROADCAST_COLLECTIVE) { in SubdivPermDebugString()
303 (col_params->instance.type == BROADCAST_COLLECTIVE && in ExecuteAsync()401 if (col_params.instance.type == BROADCAST_COLLECTIVE) { in CreateCollective()
219 cp->instance.type = BROADCAST_COLLECTIVE; in InitializeCollectiveParamsForBroadcast()313 cp->instance.type = BROADCAST_COLLECTIVE; in MakeCollectiveParams()
278 col_params_->instance.type = BROADCAST_COLLECTIVE; in Init()545 cp->instance.type = BROADCAST_COLLECTIVE; in PrepColParamsForSubdivPermsTest()797 cp->instance.type = BROADCAST_COLLECTIVE; in TEST_F()
61 case BROADCAST_COLLECTIVE: in GetCollectiveName()762 if (cp->instance.type == BROADCAST_COLLECTIVE) { in CompleteInstanceFromInitializedIRec()
79 CHECK_EQ(col_params->instance.type, BROADCAST_COLLECTIVE); in InitializeCollectiveParams()
152 case BROADCAST_COLLECTIVE: { in Enqueue()
45 BROADCAST_COLLECTIVE, enumerator