2c48e3629c
[MLIR] Adding gpu.host_register op and lower it to a runtime call.
...
Reviewed By: herhut
Differential Revision: https://reviews.llvm.org/D85631
2020-08-10 22:46:17 +02:00
2dd7a9cc2d
[MLIR] NFC: Rename mcuMemHostRegister* to mgpuMemHostRegister* to make it consistent with the other cuda-runner functions and ROCm.
...
Summary: Rename mcuMemHostRegister* to mgpuMemHostRegister*.
Reviewers: herhut
Reviewed By: herhut
Subscribers: yaxunl, mehdi_amini, rriddle, jpienaar, shauheen, antiagainst, nicolasvasilache, arpith-jacob, mgester, lucyrfox, aartbik, liufengdb, stephenneuendorffer, Joonsoo, grosul1, Kayjukh, jurahul, msifontes
Tags: #mlir
Differential Revision: https://reviews.llvm.org/D84583
2020-07-27 15:48:05 +02:00
62adfed30a
Unrank mcuMemHostRegister tensor argument.
...
Reviewers: herhut
Reviewed By: herhut
Subscribers: mehdi_amini, rriddle, jpienaar, shauheen, antiagainst, nicolasvasilache, arpith-jacob, mgester, lucyrfox, aartbik, liufengdb, stephenneuendorffer, Joonsoo, grosul1, frgossen, Kayjukh, jurahul, llvm-commits
Tags: #llvm
Differential Revision: https://reviews.llvm.org/D80118
2020-05-19 13:58:54 +02:00
b43ae21e60
Fix all-reduce int tests by host-registering memrefs.
...
Reduce amount of boiler plate to register host memory.
Summary: Fix all-reduce int tests by host-registering memrefs.
Reviewers: herhut
Reviewed By: herhut
Subscribers: clementval, mehdi_amini, rriddle, jpienaar, burmako, shauheen, antiagainst, nicolasvasilache, arpith-jacob, mgester, lucyrfox, liufengdb, Joonsoo, llvm-commits
Tags: #llvm
Differential Revision: https://reviews.llvm.org/D76563
2020-03-23 11:48:13 +01:00
c7380995f8
[MLIR] Add `and`, `or`, `xor`, `min`, `max` too gpu.all_reduce and the nvvm lowering
...
Summary:
This patch add some builtin operation for the gpu.all_reduce ops.
- for Integer only: `and`, `or`, `xor`
- for Float and Integer: `min`, `max`
This is useful for higher level dialect like OpenACC or OpenMP that can lower to the GPU dialect.
Differential Revision: https://reviews.llvm.org/D75766
2020-03-11 14:07:04 +01:00
f6790a1c63
Revert "[MLIR] Add `and`, `or`, `xor`, `min`, `max` too gpu.all_reduce and the nvvm lowering"
...
Attribution to original author got lost.
2020-03-11 14:07:04 +01:00
2eff566b07
[MLIR] Add `and`, `or`, `xor`, `min`, `max` too gpu.all_reduce and the nvvm lowering
...
Summary:
This patch add some builtin operation for the gpu.all_reduce ops.
- for Integer only: `and`, `or`, `xor`
- for Float and Integer: `min`, `max`
This is useful for higher level dialect like OpenACC or OpenMP that can lower to the GPU dialect.
Differential Revision: https://reviews.llvm.org/D75766
2020-03-10 21:09:06 +01:00
Christian Sigg