MPS conv (#822) (078c767c) · Commits · cld / ml / tvm · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tvm
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents
- Service Desk

Snippets Groups Projects

- cld
- ml
- tvm
- Commits
078c767c

Commit 078c767c authored 7 years ago by Leyuan Wang Committed by Tianqi Chen 7 years ago

Downloads
- Patches
- Plain Diff

MPS conv (#822)

parent 6731f660

No related branches found

No related tags found

No related merge requests found

Changes 10

Hide whitespace changes

Inline Side-by-side

Showing

make/contrib/mps.mk 8 additions, 2 deletions

make/contrib/mps.mk
python/tvm/contrib/mps.py 40 additions, 4 deletions

python/tvm/contrib/mps.py
python/tvm/contrib/rpc_proxy.py 2 additions, 0 deletions

python/tvm/contrib/rpc_proxy.py
src/contrib/mps/conv.mm 154 additions, 0 deletions

src/contrib/mps/conv.mm
src/contrib/mps/gemm.mm 76 additions, 82 deletions

src/contrib/mps/gemm.mm
src/contrib/mps/mps_utils.cc 0 additions, 58 deletions

src/contrib/mps/mps_utils.cc
src/contrib/mps/mps_utils.h 11 additions, 4 deletions

src/contrib/mps/mps_utils.h
src/contrib/mps/mps_utils.mm 80 additions, 0 deletions

src/contrib/mps/mps_utils.mm
src/runtime/metal/metal_device_api.mm 10 additions, 2 deletions

src/runtime/metal/metal_device_api.mm
tests/python/contrib/test_mps.py 67 additions, 23 deletions

tests/python/contrib/test_mps.py

with 448 additions and 175 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment