Searched refs:EmitNcclThunkFromMlir (Results 1 – 2 of 2) sorted by relevance
209 Status EmitNcclThunkFromMlir(MlirEmitterInput mlir_input);
3102 Status IrEmitterUnnested::EmitNcclThunkFromMlir(MlirEmitterInput input) { in EmitNcclThunkFromMlir() function in xla::gpu::IrEmitterUnnested3174 return EmitNcclThunkFromMlir<NcclAllGatherThunk, mlir::lmhlo::AllGatherOp>( in HandleAllGather()3180 return EmitNcclThunkFromMlir<NcclAllReduceThunk, mlir::lmhlo::AllReduceOp>( in HandleAllReduce()3186 return EmitNcclThunkFromMlir<NcclAllToAllThunk, mlir::lmhlo::AllToAllOp>( in HandleAllToAll()