Commit Graph

203 Commits

Author SHA1 Message Date
Tianqi Chen 8ef266063f [SCHEDULE][PASS] support storage_align of certain axis (#400)
* [SCHEDULE][PASS] support storage_align of certain axis

* fix lint
2017-08-30 13:11:18 -07:00
Tianqi Chen 0560e1569e [CODEGEN] NVPTX backend. (#392)
* [CODEGEN] NVPTX backend.

* Fix pylint

* use fix
2017-08-27 17:09:55 -07:00
Tianqi Chen 26b433b5de [TEST] Jenkinsfile (#389)
* [TEST] Jenkinsfile

* Fix wheel setup
2017-08-26 12:08:59 -07:00
Aditya Atluri 90809381e1 [RUNTIME] v2: runtime support for rocm (#386)
* v2: runtime support for rocm

* fixed coding space errors

* removed kROCM from c_runtime_api.h
2017-08-25 17:00:29 -07:00
Yizhi Liu 0ee68d7273 [APP] Android RPC (#359)
* [APP] Android RPC first version

* [APP] Android RPC build jni automatically

* [APP] Android OpenCL RPC tested on real devices

* [APP] optimize android app interface. add ndk compile tool

* add ndk compile tool

* [APP] fix android app thread crash; add android test script

* [APP] android app - show alert dialog and disconnect when error occurs

* fix ndk build script code lint

* fix ndk build default argument

* ndk script build remove shell=True. disable android app screen orientation
2017-08-25 09:02:21 -07:00
Tianqi Chen 7f1b0f5bae [SETUP] Fix python setup (#380) 2017-08-24 09:00:03 -07:00
ziheng c6ebb5a1ef [Python] Dist wheel tools (#348) 2017-08-19 20:43:11 -07:00
Tianqi Chen 090468aa53 [PASS] RewriteUnsafeSelect lowers unsafe select to condition expr (#335) 2017-08-15 22:32:43 -07:00
ziheng 25ded693ec [NNPack] Support for threadpool (#334)
* [NNPack] Support for threadpool

* fix lint

* fix lint

* Use static class function
2017-08-15 19:39:48 -07:00
ziheng 64870ffb86 [Contrib] CuDNN v7 Support (#311)
* [Contrib] CuDNN v7 Support

* Add test
2017-08-15 12:37:10 -07:00
Tianqi Chen 7196c791cb [TOPI] Isolate padding option, improve decl of depthwise/conv2d/pool (#332) 2017-08-15 11:24:07 -07:00
Xingjian Shi 760475f9a7 [TOPI] Add broadcast and reduce operators (#267)
[TOPI] Add broadcast and reduce operators
2017-08-14 19:35:40 -07:00
Tianqi Chen a59774e361 [BUILD] Simplify build process (#326) 2017-08-14 19:35:09 -07:00
Tianqi Chen a3776ba5ae [PASS][PRAGMA] Allow pragma debug_skip_region to skip region of computation (#318) 2017-08-13 15:35:38 -07:00
Tianqi Chen 79e482bc5a [PASS] Memory barrier detection, storage access lower. (#317) 2017-08-13 11:35:50 -07:00
Tianqi Chen 47eb85de3e [TEST] Upgrade gpu docker to cudnn7 (#306)
* [TEST] Upgrade gpu docker to cudnn7

* fx
2017-08-09 18:08:23 -07:00
Tianqi Chen b40d43c479 [PASS][RUNTIME] Support attr scope lift and runonce (#303) 2017-08-08 15:21:37 -07:00
Tianqi Chen 1146495f3e [RUNTIME][PASS] Allow declare vector type array (#302)
* [RUNTIME][PASS] Allow declare vector type array

* fix bcast

* [BUFFER] Enable vload/store function in buffer

* ok
2017-08-07 17:51:33 -07:00
ziheng 1e48b02f8c [NNPACK] Add nnpack.convolution (#301)
* [NNPACK] Add nnpack.convolution

* Add instrinsic

* Fix lint
2017-08-07 08:29:37 -07:00
Tianqi Chen 41768cf918 [SCHEDULE][RUNIME] Introduce pragma for additional extension hint, threadpool runtime. (#299) 2017-08-05 15:37:22 -07:00
Tianqi Chen fd96d28534 [PASS] More storage sync. (#297) 2017-08-03 22:24:51 -07:00
Tianqi Chen eefcfe1985 [PASS] Refactor thread storage sync to a common visitor (#296)
* [PASS] Refactor thread storage sync to a common visitor

* Fix the sync scope check behavior
2017-08-03 13:21:49 -07:00
Edward Z. Yang 10bc2fdfe0 Conda build recipe (#288)
* Typofix.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Probe for nvrtc in lib directory as well.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Conda build recipe for TVM.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
2017-08-01 15:23:56 -07:00
ziheng fe3b885755 [FIX] Fix doc_string of reducer (#292) 2017-08-01 10:49:59 -07:00
Yizhi Liu 657498a327 fix rpc server proxy connect (#290) 2017-08-01 09:30:12 -07:00
Tianqi Chen 7e82eb61a2 [SCHEDULE][REFACTOR] Default Fuse to outer inner, consistent to split (#289)
* [SCHEDULE] Fix fuse node order

* Make fuse order consistent with split
2017-07-31 17:57:18 -07:00
Edward Z. Yang e974443167 Build system and dynamic library fixes (#283)
* Install rules and dynamic library loading fixes.

A batch of fixes:

- Added 'install' rule to cmake and make, which installs runtime
  headers and library (libtvm_runtime).

- Added 'installdev' rule to make, which also installs the compiler
  infrastructure headers and library (libtvm)

- Added 'INSTALL_DEV' option to cmake, for toggling installation
  of compiler infrastructure headers and library

- cmake no longer builds into lib/ directory; instead all build
  products go in your build directory

- New algorithm for dynamic library loading, as described in #281.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Emit dylib on OS X

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Lint fixes.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
2017-07-31 12:23:34 -07:00
Leyuan Wang cf4e77750f [BUILD] rename build.py to avoid conflict name of build (#284)
* __init__ updated

* pull request updated

* build_module added

* typo fixed

* another typo fixed
2017-07-28 20:01:03 -07:00
Tianqi Chen 7e3d9da43c [STORAGE][BUFFER] Support access ptr for clear access pattern. (#266)
* [STORAGE][BUFFER] Support access ptr for clear access pattern.

* fix lint
2017-07-23 19:30:59 -07:00
Tianqi Chen 1f7712ae1d [LANG] Add reflection routine to construct node (#265) 2017-07-20 11:01:52 -07:00
Tianqi Chen 532ee9b7a2 [SCHEDULE] Detect useless bound early (#264)
* [SCHEDULE] Detect useless bound early

* fix
2017-07-19 15:57:31 -07:00
ziheng bdcd325686 [FIX] Fix bug and typo in rpc_server (#263)
* [FIX] Fix bug and typo in rpc_server

* [FIX] Remove unnecessary condition
2017-07-18 23:19:25 -07:00
Tianqi Chen 71cb07aee7 [RPC] IOS RPC (#261) 2017-07-18 22:20:36 -07:00
Jian Weng 01cbc61a89 [API] Prefetch schedule supported (#258)
* prefetch interface added

* prefetch python comments modified. prefetch info data structure maintained.

* start injecting prefetches. first step (domain touch) implemented.

* domain touch tested.

* Prefetch ir_mutator and ir_visitor dispatch registered.

* modify domain touched from passing a func_ref to passing a tensor

* modify domain touched from passing a func_ref to passing a tensor

* modify Tensor copy to Tensor ref

* temp commit for rebase

* debug info removed, typo fixed, ready to rebase

* prefetch flatten test add!

* roll back builtin functions to side effect functions

* lint error fixed!

* add cache line size to storage flatten argument

* forgot modifications add

* change code style to dmlc-like; get rid of can_prove, use manually compute instead

* python lint error fixed

* modify instrinsic name to pass tests

* [TEST] get rid of str(), replace them by accessing attributes

* change map to list comprehension

* redundant numpy import removed
2017-07-18 09:40:54 -07:00
Tianqi Chen 7b6427e3a8 [RPC] Allow RPCServer to run without decorator (#257) 2017-07-17 13:40:42 -07:00
Tianqi Chen 8e51af2f94 [PASS] CombineContextCall (#255) 2017-07-16 22:48:31 -07:00
ziheng 204e9cb429 [EXECUTOR] Fix bug and improve (#252)
* [EXECUTOR] Fix bug and improve

* [EXECUTOR] Enhance test case
2017-07-16 14:54:08 -07:00
Tianqi Chen 8d241b9d86 [RPC] Allow back pressure from writer (#250)
* [RPC] Allow backpressure from writer

* fix

* fix
2017-07-15 18:23:20 -07:00
ziheng c6d4f5af20 [EXECUTOR] Enable load executor remotely (#245)
* [EXECUTOR] Enable load executor remotely

* [EXECUTOR] Pipeline

* Pass bytearray directly

* Enable load dynamic library in rpc_server.py

* Fix

* lint

* Return Module from remote side directly

* Remove unused header file

* Fix

* fix
2017-07-15 13:03:30 -07:00
Tianqi Chen 3f3bf29d5a [DOC] Make range related function consistent (#249) 2017-07-15 10:43:20 -07:00
Tianqi Chen 2ab0bfb589 [RENAME] nvcc_compiler->nvcc, cc_compiler->cc, metal_compiler->xcode (#248) 2017-07-15 09:41:44 -07:00
Tianqi Chen 7bcb3f538b [REFACTOR] collections->container, RPC returns func, time_evaluator r… (#244)
* [REFACTOR] collections->container, RPC returns func, time_evaluator returns struct

* fix executor
2017-07-13 22:30:22 -07:00
Tianqi Chen c324494f1a [RUNTIME][RPC] Change RPCServer to Event Driven Code (#243)
* [RUNTIME][RPC] Change RPCServer to Event Driven Code

* fix
2017-07-13 18:44:33 -07:00
Tianqi Chen 0a07411d54 [JS][WEB][BACKEND] Javascript(webassembly) backend. (#239) 2017-07-10 18:23:47 -07:00
Tianqi Chen adc06e6fbb [RUNTIME][ABI] Remove TVMValue as argument, use address (#236) 2017-07-09 18:02:56 -07:00
Tianqi Chen cd29c18c34 [RUNTIME][ABI] Flat structure arguments (#232) 2017-07-08 16:26:36 -07:00
Tianqi Chen 9d84cb0777 [RUNTIME] Add workspace pool (#229)
* [RUNTIME] Add workspace pool

* fix doc

* fix the free list

* avoid zero size
2017-07-07 23:19:57 -07:00
ziheng 5cdc860492 [TAG] Fix signature of decorated function (#228)
* [TAG] Fix signature of decorated function

* Add dep
2017-07-07 16:02:35 -07:00
Tianqi Chen 72fcd4e6ee [RUNTIME] Add System Lib (#227)
* [RUNTIME] Add System Lib

* lint

* lint

* fix compile
2017-07-07 12:16:37 -07:00
Tianqi Chen 825566ccff [SCHEDULE] tensorize (#223) 2017-07-06 13:56:39 -07:00