enable rocm target for topi/recipes. add timing util to gemm test. (#554) (c3cac464) · 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
c3cac464

Commit c3cac464 authored 7 years ago by masahi Committed by Tianqi Chen 7 years ago

Downloads
- Patches
- Plain Diff

enable rocm target for topi/recipes. add timing util to gemm test. (#554)

parent 592a1f65

No related branches found

No related tags found

No related merge requests found

Changes 3

Hide whitespace changes

Inline Side-by-side

Showing

topi/recipe/conv/depthwise_conv2d_test.py 16 additions, 14 deletions

topi/recipe/conv/depthwise_conv2d_test.py
topi/recipe/conv/test_conv2d_hwcn_map.py 4 additions, 4 deletions

topi/recipe/conv/test_conv2d_hwcn_map.py
topi/recipe/gemm/cuda_gemm_square.py 14 additions, 5 deletions

topi/recipe/gemm/cuda_gemm_square.py

with 34 additions and 23 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