shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
340,317 Commits 1,713 Branches 298 Tags
Commit Graph

10 Commits

Author SHA1 Message Date
Tres Popp
9a52ea5cf9 Create a gpu.module operation for the GPU Dialect. 
Summary:
This is based on the use of code constantly checking for an attribute on
a model and instead represents the distinct operaion with a different
op. Instead, this op can be used to provide better filtering.

Reverts "Revert "[mlir] Create a gpu.module operation for the GPU Dialect.""

This reverts commit ac446302ca4145cdc89f377c0c364c29ee303be5 after
fixing internal Google issues.

This additionally updates ROCDL lowering to use the new gpu.module.

Reviewers: herhut, mravishankar, antiagainst, nicolasvasilache

Subscribers: jholewinski, mgorny, mehdi_amini, jpienaar, burmako, shauheen, csigg, arpith-jacob, mgester, lucyrfox, aartbik, liufengdb, llvm-commits, mravishankar, rriddle, antiagainst, bkramer

Tags: #llvm

Differential Revision: https://reviews.llvm.org/D72921
 2020-01-21 14:05:03 +01:00
Benjamin Kramer
a2cd4fe6bf Unbreak the mlir build after 202ab273e6eca134b69882f100c666fcd3affbcf 2020-01-13 19:18:43 +01:00
Julian Gross
202ab273e6 [mlir] Added missing GPU lowering ops. 
Summary:
This diff adds missing GPU lowering ops to MLIR.

Reviewers: herhut, pifon2a, ftynse

Tags: #pre-merge_beta_testing, #llvm

Differential Revision: https://reviews.llvm.org/D72439
 2020-01-13 17:10:54 +01:00
Mehdi Amini
56222a0694 Adjust License.txt file to use the LLVM license 
PiperOrigin-RevId: 286906740
 2019-12-23 15:33:37 -08:00
Alex Zinenko
b8dc3fd812 Rename CLI flags -lower-gpu-ops-to-*-ops to -convert-gpu-to-* 
This makes the flags consistent with the naming scheme used elsewhere in the
codebase for dialect conversions.

PiperOrigin-RevId: 281027517
 2019-11-18 02:43:10 -08:00
Alexander Belyaev
d2ce435dba Add custom lowering of ExpOp for NVVM and ROCM. 
PiperOrigin-RevId: 276440911
 2019-10-24 01:41:57 -07:00
Christian Sigg
b74af4aa5c Unify GPU op definition names with other dialects. 
Rename GPU op names from gpu_Foo to GPU_FooOp.

PiperOrigin-RevId: 275882232
 2019-10-21 11:10:56 -07:00
Deven Desai
d064469f6f Moving the GPUIndexIntrinsicOpLowering template to a common location 
The GPUIndexIntrinsicOpLowering template is currently used by the code in both the GPUToNVVM and GPUToROCDL dirs.
Moving it to a common location to remove code duplication.

Closes tensorflow/mlir#163

COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/163 from deven-amd:deven-refactor-gpu-index-ops-lowering b8dc2a5f5353df196039b6ff2ad42106028693ed
PiperOrigin-RevId: 272863297
 2019-10-04 06:20:05 -07:00
Christian Sigg
85dcaf19c7 Fix typos, NFC. 
PiperOrigin-RevId: 272851237
 2019-10-04 04:37:53 -07:00
Deven Desai
e81b3129b4 [ROCm] Adding pass to lower GPU Dialect to ROCDL Dialect. 
This is a follow-up to the PRtensorflow/mlir#146 which introduced the ROCDL Dialect. This PR introduces a pass to lower GPU Dialect to the ROCDL Dialect. As with the previous PR, this one builds on the work done by @whchung, and addresses most of the review comments in the original PR.

Closes tensorflow/mlir#154

COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/154 from deven-amd:deven-lower-gpu-to-rocdl 809893e08236da5ab6a38e3459692fa04247773d
PiperOrigin-RevId: 272390729
 2019-10-02 01:50:30 -07:00