Add rocblas_sgemm_strided_batched impl. (#6579)
authorChris Sullivan <csullivan@octoml.ai>
Tue, 29 Sep 2020 09:59:54 +0000 (02:59 -0700)
committerGitHub <noreply@github.com>
Tue, 29 Sep 2020 09:59:54 +0000 (18:59 +0900)
commit22ad0dd4cc21f44a605ec2552945bd5f068a6959
tree0aa01ed9fa54fb8dd154a576aef1c360de1b716e
parentcab4ce19dd70e21232697986f0b717434f0cce4d
Add rocblas_sgemm_strided_batched impl. (#6579)
include/tvm/topi/contrib/rocblas.h
python/tvm/contrib/rocblas.py
python/tvm/relay/op/strategy/rocm.py
python/tvm/topi/rocm/__init__.py
python/tvm/topi/rocm/batch_matmul.py [new file with mode: 0644]
src/runtime/contrib/rocblas/rocblas.cc
tests/python/contrib/test_rocblas.py