Searched refs:procInfo (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/ |
| H A D | TestLinalgTransforms.cpp | 254 SmallVector<ProcInfo, 2> procInfo(count); in getGpuProcIds() local
258 procInfo[count - 1 - i] = {b.create<IdOp>(loc, indexType, dim), in getGpuProcIds()
261 return procInfo; in getGpuProcIds()
270 cyclicNprocsEqNiters.procInfo = in fillTileAndDistributePatterns()
287 cyclicNprocsGeNiters.procInfo = in fillTileAndDistributePatterns()
304 cyclicNprocsDefault.procInfo = in fillTileAndDistributePatterns()
322 cyclicNprocsMixed1.procInfo = getGpuProcIds<gpu::BlockIdOp, gpu::GridDimOp>; in fillTileAndDistributePatterns()
339 cyclicNprocsMixed2.procInfo = getGpuProcIds<gpu::BlockIdOp, gpu::GridDimOp>; in fillTileAndDistributePatterns()
356 cyclicNprocsMixed3.procInfo = getGpuProcIds<gpu::BlockIdOp, gpu::GridDimOp>; in fillTileAndDistributePatterns()
373 cyclicNprocsEqNiters.procInfo = in fillTileAndDistributePatterns()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 510 SmallVector<ProcInfo, 4> procInfo; in doit() local
523 procInfo = distributionOptions->procInfo(b, loc, parallelLoopRanges); in doit()
552 for (auto it : llvm::zip(loops, procInfo, distributionMethod)) in doit()
762 SmallVector<ProcInfo, 2> procInfo = in doit() local
763 options.procInfo(b, loc, parallelLoopRanges); in doit()
766 if (index >= procInfo.size()) in doit()
770 updateBoundsForCyclicDistribution(b, loc, procInfo[index].procId, in doit()
771 procInfo[index].nprocs, lbsStorage[i], in doit()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 357 ProcInfoCallBackFn procInfo; member
|