Daniil Fukalov 406d03bfaf Improve ROCm support. (#780) ...

- updates to support ROCm 5.2
- workarounds in tests where NV tools were used unconditionally
- implemented `get_num_blocks()` and `add_memfence()` for AMD GPU
- backported from history some atomics
- added bf16 support
- minor warnings cleanup
- added dockerfile to run on a ROCm enabled machine

Co-authored-by: B1tway <andrew.shukshov@gmail.com>
Co-authored-by: Andrey Shukshov <36711069+B1tway@users.noreply.github.com>

2022-10-14 11:33:42 -07:00

..

analysis

[ROCM] enable matmul(dot) and others (#391)

2021-12-13 12:28:15 -08:00

selection

Improve ROCm support. (#780)

2022-10-14 11:33:42 -07:00

transform

[CODEGEN] Various bugfixes that make it possible to fuse RNG in a matmul epilogue (#356)

2021-10-24 02:30:46 -07:00

pass.cc

[ROCM] enable matmul(dot) and others (#391)

2021-12-13 12:28:15 -08:00

target.cc

Improve ROCm support. (#780)

2022-10-14 11:33:42 -07:00