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
0560e156

Commit 0560e156 authored 7 years ago by Tianqi Chen Committed by GitHub 7 years ago

Downloads
- Patches
- Plain Diff

[CODEGEN] NVPTX backend. (#392)

* [CODEGEN] NVPTX backend.

* Fix pylint

* use fix

parent efafa1a0

No related branches found

No related tags found

Loading

Changes 8

Hide whitespace changes

Inline Side-by-side

Showing

python/tvm/_ffi/ndarray.py 4 additions, 1 deletion

python/tvm/_ffi/ndarray.py
src/codegen/llvm/codegen_llvm.cc 70 additions, 17 deletions

src/codegen/llvm/codegen_llvm.cc
src/codegen/llvm/codegen_llvm.h 25 additions, 10 deletions

src/codegen/llvm/codegen_llvm.h
src/codegen/llvm/codegen_nvptx.cc 167 additions, 0 deletions

src/codegen/llvm/codegen_nvptx.cc
src/codegen/llvm/llvm_common.cc 41 additions, 39 deletions

src/codegen/llvm/llvm_common.cc
src/runtime/module.cc 2 additions, 0 deletions

src/runtime/module.cc
tests/python/integration/test_gemm.py 1 addition, 0 deletions

tests/python/integration/test_gemm.py
tests/python/unittest/test_codegen_device.py 1 addition, 1 deletion

tests/python/unittest/test_codegen_device.py

with 311 additions and 68 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