Commit Graph

311 Commits

Author SHA1 Message Date
Tianqi Chen 3d67ea17d2 [RPC] support tracker in proxy (#1082) 2018-04-05 21:44:07 -07:00
Tianqi Chen 79fc66724d [RPC] Tracker status query (#1081) 2018-04-05 13:47:48 -07:00
Tianqi Chen 6bd8dbc764 [RPC] Refactor, introduce tracker (#1080)
* [RPC] Refactor, introduce tracker

* [RPC] Change RPC hand shake convention, always get remote key.

* fix lint
2018-04-04 20:40:41 -07:00
Lianmin Zheng 9fb260568d mix fix (#1079) 2018-04-04 14:31:24 -07:00
alex-weaver aaf7ff04e3 Move BuildConfig context stack to C++ (#1025) 2018-04-01 10:42:54 -07:00
Lianmin Zheng 8065f2e078 support string option when create cuda/rocm/... target (#1071) 2018-03-31 15:48:05 -07:00
Tianqi Chen c29205da87 [DOCS] Try upgrade build (#1066) 2018-03-30 17:46:10 -07:00
Ding cc7a8fcf99 [TOPI] Overload operators of Tensor when TOPI is imported (#1029) 2018-03-27 12:57:20 -04:00
Tianqi Chen e15aae2b3e [SCHEDULE][PASS] Enable Warp memory and lower to shuffle (#1050)
* [SCHEDULE][PASS] Enable Warp memory and lower to shuffle

* OpenCL dispatches for now to intel shuffle
2018-03-26 10:43:40 -04:00
Tianqi Chen cc71d505a0 [CONTRIB] Patch nnvcc to generate error when build the empty result (#1049) 2018-03-25 15:07:57 -04:00
Tianqi Chen 1ab05faffa [CYTHON] Fix exception propagation for cython3 (#1046) 2018-03-24 18:34:53 -07:00
alex-weaver 6292204e94 Implement C++ registry to back Python target.generic_func (#892) 2018-03-19 00:17:25 -07:00
Tianqi Chen 6bcf95f2e4 [CONTRIB] windows compatiblity (#1009) 2018-03-15 19:16:45 -07:00
Ding 34e31c4439 [PASS] Add VerifyMemory pass and test cases (#410) (#993) 2018-03-13 08:52:21 -07:00
Tianqi Chen 89d5e552d5 [RUNTIME] Update graph runtime to rely on smarter planner, add get_input (#990) 2018-03-10 15:32:32 -08:00
Tianqi Chen bcb3bef579 [CYTHON] Correct backtrace print for python3 (#989) 2018-03-10 14:25:45 -08:00
Tianqi Chen a6e8fe6c70 [FFI] Fix global free destruction (#985) 2018-03-10 09:40:52 -08:00
Tianqi Chen df1b4f6461 [IOS] Improve the iOS RPC with exclusive filesys lock (#981) 2018-03-09 11:12:57 -08:00
libing4752 38274115a9 enhance access_ptr that args can support Expr (#970) 2018-03-07 20:04:21 -08:00
Leyuan Wang 078c767cda MPS conv (#822) 2018-03-06 17:51:13 -08:00
eqy 3e543bd055 prevent starting of RPC server w/o RPC support (#962)
* prevent starting of RPC server w/o RPC support

* fix indent
2018-03-05 13:00:22 -08:00
Tianqi Chen 0d6cf0c870 Update function.py 2018-03-02 15:39:52 -08:00
Tianqi Chen 0e6370db4f explicit import testing (#956)
* explicit import testing

* Enable init api for extension modules
2018-03-02 15:19:13 -08:00
Atsushi Nukariya 549aa16555 Fix a typo for Target class (#951) 2018-03-01 21:07:35 -08:00
Tianqi Chen 413e2b7a2e Better error message handling for contrib (#946)
* Better error message handling for contrib

* fix lint

* fix testcase

* fix test
2018-03-01 10:59:58 -08:00
Pariksheet Pinjari e3f6938a22 Spelling mistake corrected (#945) 2018-02-28 19:42:50 -08:00
Lianmin Zheng 07cc21f106 add exclusive mode for rpc server (#941) 2018-02-28 08:54:58 -08:00
Zhixun Tan 589a26515f Add test case: Create a static WebGL library and run it in the browser. (#932)
* Add test case: Create a static WebGL library and run it in the browser.

* Add documentation for loadModuleFromFile

* Modify emscripten.createjs
2018-02-27 20:20:16 -08:00
Tianqi Chen 1c97eaf622 MXNet NDArray bridge. (#930)
* MXNet NDArray bridge.
Support convert a tvm Function as MXNet's async NDArray function.

* fix lint

* update comment
2018-02-25 11:45:45 -08:00
Jammy Zhou 60d42a97a8 Fix a typo for function registration (#927) 2018-02-24 16:34:03 -08:00
Tianqi Chen d99bcaf156 [EXT] Allow easy extraction of extern module (#926) 2018-02-23 10:24:17 -08:00
Tianqi Chen 433756b9fd Revert "[RUNTIME] Refactor extension type handling, now it is header only (#924)" (#925)
This reverts commit 12d15704d7f5d30cff7540f1fd16be64c6baca68.
2018-02-22 20:18:08 -08:00
Tianqi Chen a6b4a219e2 [RUNTIME] Refactor extension type handling, now it is header only (#924)
* [RUNTIME] Refactor extension type handling, now it is header only
2018-02-22 18:26:32 -08:00
libing4752 61cdf903dd [SCHEDULE] Add factor_axis to rfactor (#895) 2018-02-22 09:28:23 -08:00
Tianqi Chen 12d1ab5ae2 [RUNTIME] More reliable runtime only detection (#914)
* [RUNTIME] More reliable runtime only detection

* fix lint
2018-02-20 16:56:59 -08:00
alex-weaver 523b6a6bec Convert BuildModule to use TVM node system (#879)
* Make python BuildConfig serializable/deserializable to/from string

* Make C++ BuildConfig serializable/deserializable to/from string

* Revert "Make python BuildConfig serializable/deserializable to/from string"

This reverts commit a5e1fb3ff63a161cc0d63475d2a32816cc4c3666.

* Revert "Make C++ BuildConfig serializable/deserializable to/from string"

This reverts commit ec0c2c54543050fe6f264d06eebff33dee70370b.

* Converted BuildConfig to use TVM node system

* Fix lint

* Fix lint

* Added code to set node attributes through the C API

* Fixed bug in build_config()

* Fix lint

* Fix lint

* Fix test errors

* Reduced scope of node __setattr__ to apply only to BuildConfig

* Fix lint

* Fix lint

* Changed python BuildConfig to be immutable, with values set once on construction.

* Fix lint

* Fix C++ test

* Fixed BuildConfig setting python-side args

* Fix lint

* Removed dependency on reflection.cc to construct BuildConfig (allow use in runtime library)

* Fix lint

* Revert "Fix lint"

This reverts commit 16ed6d7a1ca5e551b035bad46e8361ea487cd45b.

* Revert "Removed dependency on reflection.cc to construct BuildConfig (allow use in runtime library)"

This reverts commit 43817c97a2ee045791e0c031d962fa97636ce8f6.

* Avoid accessing BuildConfig when using runtime lib

* Fix missing import

* Fix error running under cython (root cause: node handle is not valid until after __init__ has returned, so cannot call __dir__ during __init__

* Fix error where BuildConfig._node_defaults was not copied in build_config()

* Fix lint

* Fix lint

* Fix lint

* Fix lint

* Add comments to python BuildConfig
2018-02-10 12:39:34 -08:00
Zhixun Tan 0a410a39dc [WIP] Add OpenGL topi. (#836)
[TOPI][GL] OpenGL topi.
2018-02-07 12:57:00 -08:00
Lianmin Zheng fb556ef4b8 [CONTRIB] add peak test (#878)
* add peak test

* fix error for lanes=16

* update doc

* fix names

* fix names
2018-02-06 17:51:03 -08:00
Tianqi Chen 7d2654c2ba [CODEGEN] Fix vector element access in metal (#872) 2018-02-04 23:06:46 -08:00
yuruofeifei 2e94a4b55f [TOPI] Add compute for more operators (#849)
* [TOPI] Add compute for more operators

* Remove device except llvm

* Address comments

* Remove matmul compute

* Add outtype to boolean operator

* Address coments
2018-02-02 14:33:29 -08:00
Tianqi Chen 79d503fd3b [BACKEND] Vulkan Runtime and SPIRV Codegen (#861)
* [BACKEND] Vulkan Runtime and SPIRV Codegen

* fix doc
2018-02-01 22:24:57 -08:00
Tianqi Chen bf71582cc9 [RELEASE] Release note for 0.2 (#853) 2018-01-31 11:58:43 -08:00
Tianqi Chen ec9f38ec44 [TOPI] Fix cpp library dependency on MAC (#852) 2018-01-31 10:20:57 -08:00
alex-weaver f280f23a3d Porting schedules (except convolutions) to C++ (#763)
* Ported injective schedules to C++. Added some elementwise ops.

* Fix lint errors

* Added reduction ops and schedules

* Fix lint errors

* Fix lint errors

* Fix lint errors

* Added transform ops

* Fix lint errors

* Fix lint errors

* Added softmax, log_softmax, leaky_relu and flatten ops.
Fixed issue where TVM_DECLARE_INTRIN_UNARY used the PureExtern flag
instead of PureIntrinsic.
Added softmax CUDA schedule.

* Fix lint

* Fix lint

* Added binary_dense, batch_norm_inference, dense, dilate, scale_shift_*,
global_pool and pool ops.
Extended pad to allow specifying pad_value.
Fixed issue where pad would throw if padding was zero in all dimensions.

* Fix lint

* Fix lint

* Added CUDA schedules for dense, pool and global_pool

* Added extern schedules for generic and CUDA

* Fix lint

* Added x86 binary schedules

* Fix lint

* Added rocm dense schedule. Added rocBLAS and cuBLAS support to dense ops

* Added pow ops. Added x86 default and injective schedules

* Fix lint

* Fix lint

* Fix lint

* Fix lint

* Fix lint

* Fix indent

* Removed schedules directory

* Changed left_shift, right_shift to operators. Changed pad_value in pad() to remove pointer usage

* Fixed usage of pad in nn/pooling.h. Fixed declaration of operator>>

* Fixed comments for shift operators

* Added comments to utility functions

* Added TOPI C++ library, exporting broadcast_add op

* Fix lint

* Share libinfo.py with TVM

* Fix lint

* Add other broadcast ops

* Fix lint

* Fix imports in topi

* Fix lib names

* Fixed build issue where windows builds don't apply correct definitions

* Removed TVM_EXPORTS from topi library

* Attempted CI build fix

* Add topi lib to tvm_multilib

* Fix Jenkinsfile

* Added TOPI build target to Makefile

* Fix nn op namespaces.

* Fix lint

* Renamed TOPI lib to libtvm_topi

* Removed _ffi/base.py

* Remove _ffi from topi, now shared with tvm.

* Make libtvm_topi loading optional

* Fix compiler warnings

* Fix lint

* Fix lint

* Fix lint

* Fix build error by making new libs argument to Target optional

* Added C++ Target type interop. Added registration of remaining C++ ops and schedules. Added test of broadcast ops

* Fix lint

* Fix lint

* Fix compile error

* Fix compiler warnings

* Fix compiler warnings

* Fixed int vector interop. Fixed argmin incorrectly invoking argmax. Fixed corner case in default schedules of attempting to fuse 0 length axes. Added tests for reduce ops.

* Refactored reduce builders

* Fixed typos in topi.cc. Added basic test.

* Fixed padding size error. Added dense, dilate, pooling tests

* Fixed issue where clip would output a different dtype to the input. Added split_sections op to cover the other mode of the python split op. Added tests.

* Changed extension type numbers to avoid clash with NNVM

* Fix lint

* Fix compiler warnings

* Removed use of std::vector from the public TOPI API

* Fix lint

* Add TOPI C++ tests to CI

* Fixed detail namespacing. Improved comments.
2018-01-27 21:57:13 -08:00
Siva eb8077fff8 [DEBUG] get_node_output : To retrieve out put of any node - for debug purpose. (#820) 2018-01-27 21:50:04 -08:00
Tianqi Chen ba8d00c2da [TIMER] Enhance time evaluator to create multiple results (#830) 2018-01-27 10:45:28 -08:00
kun-zh 293dac397b support using pointer with an original offset (#826)
* when there is no intrin func, using body for initialization. For issue 714.

* Refine code per review comments, and add a test case.

* Fix lint issues.

* Re-organize the tensorize test cases, and add a new case for none-reset
mode.

* Fix a typo.

* Delete the unit case because merged it into test_schedule_tensorize.py already.

* always use new tensor in its stage when rewrite for cache read

* revert previous changes to sync up with master

* support using the ptr with an original offset

* update test case and fix CI error
2018-01-27 09:10:42 -08:00
Clouds d1cdb623c3 Update setup.py (#803)
fix errors when running `python3 setup.py sdist bdist_wheel`
2018-01-22 10:00:08 -08:00
Zhixun Tan 589831df65 [WIP] WebGL Backend (#672)
Basic WebGL Backend
2018-01-20 13:51:34 -08:00
xqdan d4a46898e0 Support dump ir for each pass (#693) (#791)
* Support dump ir for each pass(#693)

* expose DumpIR

* fix comments

* fix comments
2018-01-19 19:57:39 -08:00