| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | MemoryPromotion.cpp | 51 if (rank < GPUDialect::getNumWorkgroupDimensions()) { in insertCopyLoops()
52 unsigned extraLoops = GPUDialect::getNumWorkgroupDimensions() - rank; in insertCopyLoops()
92 GPUDialect::getNumWorkgroupDimensions())))) { in insertCopyLoops()
165 int workgroupMemoryAddressSpace = gpu::GPUDialect::getWorkgroupAddressSpace(); in promoteToWorkgroupMemory()
|
| D | KernelOutlining.cpp | 160 outlinedFunc.setAttr(gpu::GPUDialect::getKernelFuncAttrName(), in outlineKernelFuncImpl()
272 getOperation().setAttr(gpu::GPUDialect::getContainerModuleAttrName(), in runOnOperation()
|
| D | AllReduceLowering.cpp | 155 gpu::GPUDialect::getWorkgroupAddressSpace(); in createWorkgroupBuffer()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 34 bool GPUDialect::isKernel(Operation *op) { in isKernel()
39 void GPUDialect::initialize() { in initialize()
47 Type GPUDialect::parseType(DialectAsmParser &parser) const { in parseType()
62 void GPUDialect::printType(Type type, DialectAsmPrinter &os) const { in printType()
68 LogicalResult GPUDialect::verifyOperationAttribute(Operation *op, in verifyOperationAttribute()
108 GPUDialect::getKernelFuncAttrName())) in verifyOperationAttribute()
110 << GPUDialect::getKernelFuncAttrName() << "' attribute"; in verifyOperationAttribute()
478 GPUDialect::getContainerModuleAttrName())) in verify()
481 GPUDialect::getContainerModuleAttrName() + "' attribute"); in verify()
649 result.addAttribute(GPUDialect::getKernelFuncAttrName(), in parseGPUFuncOp()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/GPU/ |
| D | CMakeLists.txt | 2 IR/GPUDialect.cpp
|
| /external/llvm-project/mlir/include/mlir/ |
| D | InitAllDialects.h | 49 gpu::GPUDialect, in registerAllDialects()
|
| /external/llvm-project/mlir/lib/Conversion/ |
| D | PassDetail.h | 23 class GPUDialect; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | BUILD | 51 "@llvm-project//mlir:GPUDialect",
84 "@llvm-project//mlir:GPUDialect",
|
| D | passes.cc | 260 new_kernel->setAttr(mlir::gpu::GPUDialect::getKernelFuncAttrName(), in runOnFunction()
|
| /external/llvm-project/mlir/test/lib/Dialect/SPIRV/ |
| D | TestEntryPointAbi.cpp | 47 if (!gpu::GPUDialect::isKernel(gpuFunc) || gpuFunc.getAttr(attrName)) in runOnOperation()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToNVVM/ |
| D | LowerGpuOpsToNVVMOps.cpp | 122 if (type.getMemorySpace() != gpu::GPUDialect::getPrivateAddressSpace()) in runOnOperation()
150 target.addIllegalDialect<gpu::GPUDialect>(); in configureGpuToNVVMConversionLegality()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 175 registry.insert<AffineDialect, gpu::GPUDialect, linalg::LinalgDialect, in getDependentDialects()
183 gpu::GPUDialect, scf::SCFDialect, LmhloDialect>(); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPUPass.cpp | 54 target.addLegalDialect<gpu::GPUDialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Target/LLVMIR/ |
| D | ConvertToNVVMIR.cpp | 80 if (!gpu::GPUDialect::isKernel(func)) in translateModuleToNVVMIR()
|
| D | ConvertToROCDLIR.cpp | 90 gpu::GPUDialect::getKernelFuncAttrName())) in translateModuleToROCDLIR()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/ |
| D | ConvertGPUToSPIRV.cpp | 241 if (!gpu::GPUDialect::isKernel(funcOp)) in matchAndRewrite()
271 newFuncOp.removeAttr(Identifier::get(gpu::GPUDialect::getKernelFuncAttrName(), in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToROCDL/ |
| D | LowerGpuOpsToROCDLOps.cpp | 72 target.addIllegalDialect<gpu::GPUDialect>(); in runOnOperation()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgCodegenStrategy.cpp | 37 gpu::GPUDialect, in getDependentDialects()
|
| /external/tensorflow/third_party/mlir/ |
| D | test.BUILD | 260 "@llvm-project//mlir:GPUDialect",
328 "@llvm-project//mlir:GPUDialect",
|
| D | BUILD | 1801 name = "GPUDialect",
1809 "include/mlir/Dialect/GPU/GPUDialect.h",
1856 ":GPUDialect",
1893 ":GPUDialect",
1932 ":GPUDialect",
1955 ":GPUDialect",
2014 ":GPUDialect",
2041 ":GPUDialect",
2065 ":GPUDialect",
2111 ":GPUDialect",
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | GPUOpsLowering.h | 52 gpu::GPUDialect::getWorkgroupAddressSpace()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | BUILD | 106 "@llvm-project//mlir:GPUDialect",
|
| D | tf_kernel_to_llvm_pass.cc | 268 gpu::GPUDialect, tf_framework::TFFrameworkDialect, in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | BUILD | 65 "@llvm-project//mlir:GPUDialect",
|
| /external/llvm-project/mlir/include/mlir/Conversion/ |
| D | Passes.td | 251 let dependentDialects = ["gpu::GPUDialect"];
263 let dependentDialects = ["AffineDialect", "gpu::GPUDialect"];
|