Home
last modified time | relevance | path

Searched refs:ShuffleMode (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/
H A DLowerGpuOpsToNVVMOps.cpp46 static NVVM::ShflKind convertShflKind(gpu::ShuffleMode mode) { in convertShflKind()
48 case gpu::ShuffleMode::XOR: in convertShflKind()
50 case gpu::ShuffleMode::UP: in convertShflKind()
52 case gpu::ShuffleMode::DOWN: in convertShflKind()
54 case gpu::ShuffleMode::IDX: in convertShflKind()
104 if (op.mode() == gpu::ShuffleMode::UP) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp341 shuffleType, value, offset, activeWidth, gpu::ShuffleMode::XOR); in createSubgroupReduce()
364 gpu::ShuffleMode::XOR); in createSubgroupReduce()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Vector/
H A DTestVectorTransforms.cpp814 /*mode=*/gpu::ShuffleMode::XOR) in warpReduction()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/
H A DGPUDialect.cpp791 int32_t offset, int32_t width, ShuffleMode mode) { in build()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/
H A DGPUOps.td731 def GPU_ShuffleMode : I32EnumAttr<"ShuffleMode",
773 "ShuffleMode":$mode)>
/llvm-project-15.0.7/llvm/lib/Transforms/Vectorize/
H A DSLPVectorizer.cpp362 enum ShuffleMode { Unknown, Select, Permute }; in isFixedVectorShuffle() enum
363 ShuffleMode CommonShuffleMode = Unknown; in isFixedVectorShuffle()