* Remove all attr::storage_scope usage
* pyformat
* fixed VTA tests
* Update TIR text printer to print storage_scope on allocate
* print storage scope in AllocateNode ReprPrinter
* Fixed accidently removed scope tag check
* remove unused function
Co-authored-by: masa <masa@pop-os.localdomain>
* Add storage scope to ProducerRealize, always create a buffer with scope
* update schedule_ops.cc
* update schedule_postproc_to_primfunc.cc
* restore more realize_scope
This reverts commit b66c3baa54feeb8e34016713a1be21802b3296bf.
* make the default scope be "" instead of None in ir builder
* restore realize_scope visit in storage_flatten.cc
* update storage_access.cc
* make sure buffer var is of PointerType in ir builder
This reverts commit e650b6c24cabd52a073064e51c2e4fee816e88fd.
* enforce default storage scope of global
* added remap pass but does not work yet
* fixed all reduce issue
This reverts commit 8e20003c5325085ed22ee57180aca18644b3b5ab.
* simplify
* trying mitigation for aot test
* merge remaining changes from initial branch
* remove use of attr::storage_scope from codegen
* restore a visit to AttrStmt with attr::storage_scope in storage_rewrite
* disable check
* lint fix
* revert default scope to ""
* format
* fix volatile access to shared mem in lower all reduce
* fixed gpu coorporative load/store test
* pass storage scope to PointerType in tvm script parser
This reverts commit 99cfb9d18781dcfdea169d920450f9063ab18b6b.
* fixed tvmscript roundtrip test
* fixed tir flatten buffer test
* fixed test_tir_transform_hoist_if.py
* use storage scope global by default in aot_executor_codegen.cc
* add missing default storage scope in create_primfunc.cc
* restore StorageInfo struct in llvm backend
* UpdateStorageScope -> WithStorageScope
* fixed lower warp memory test
* GetStorageScope -> GetPtrStorageScope
* Enable storage scope invariant check in AttrStmt constructor
* remove GetPtrStorageScope and WithStorageScope from public header
* move RemapStorageScope to its own file
* add more method to RemapStorageScope
* update lower_thread_allreduce to use RemapStorageScope
* RemapStorageScope -> UpdatePointerStorageScope
* remove realize_scope from hybrid script
* removed realize_scope in schedule_ops
* remove realize_scope from schedule_postproc_to_primfunc
* remove remaining realize_scope usage from schedule_ops.cc
* remove realize_scope usage from storage_flatten.cc
* fixed test_tir_transform_lower_warp_memory.py following realize_scope removal
* Add storage scope to ProducerRealize, always create a buffer with scope
* update schedule_ops.cc
* update schedule_postproc_to_primfunc.cc
* restore more realize_scope
This reverts commit b66c3baa54feeb8e34016713a1be21802b3296bf.
* make the default scope be "" instead of None in ir builder
* restore realize_scope visit in storage_flatten.cc
* update storage_access.cc
* make sure buffer var is of PointerType in ir builder
This reverts commit e650b6c24cabd52a073064e51c2e4fee816e88fd.
* enforce default storage scope of global
* added remap pass but does not work yet
* fixed all reduce issue
This reverts commit 8e20003c5325085ed22ee57180aca18644b3b5ab.
* simplify
* trying mitigation for aot test
* merge remaining changes from initial branch
* remove use of attr::storage_scope from codegen
* restore a visit to AttrStmt with attr::storage_scope in storage_rewrite
* disable check
* lint fix
* revert default scope to ""
* format
* fix volatile access to shared mem in lower all reduce
* fixed gpu coorporative load/store test
* pass storage scope to PointerType in tvm script parser
This reverts commit 99cfb9d18781dcfdea169d920450f9063ab18b6b.
* fixed tvmscript roundtrip test
* fixed tir flatten buffer test
* fixed test_tir_transform_hoist_if.py
* use storage scope global by default in aot_executor_codegen.cc
* add missing default storage scope in create_primfunc.cc
* restore StorageInfo struct in llvm backend
* UpdateStorageScope -> WithStorageScope
* fixed lower warp memory test
* GetStorageScope -> GetPtrStorageScope
* Enable storage scope invariant check in AttrStmt constructor
* remove GetPtrStorageScope and WithStorageScope from public header
* move RemapStorageScope to its own file
* add more method to RemapStorageScope
* update lower_thread_allreduce to use RemapStorageScope
* RemapStorageScope -> UpdatePointerStorageScope
* remove realize_scope from hybrid script
* removed realize_scope in schedule_ops
* remove realize_scope from schedule_postproc_to_primfunc
* remove remaining realize_scope usage from schedule_ops.cc
* remove realize_scope usage from storage_flatten.cc
* fixed test_tir_transform_lower_warp_memory.py following realize_scope removal
* Address comments
* Remove blank line diff
Co-authored-by: Masahiro Masuda <masahi@129@gmail.com>
Co-authored-by: masa <masa@pop-os.localdomain>
With either the ci_lint docker image, or the matched version of
pylint==2.4.4, I got two lint errors running locally that didn't show
up in the CI. Fixing them.
Co-authored-by: Eric Lunderberg <elunderberg@octoml.ai>
* [TVMSCRIPT] add float type support in script function
* [TVMSCRIPT] add more type support in script function parameter
Co-authored-by: honghua.cao <honghua.cao@streamcomputing.com>
* Add workspace size calculation for primfuncs
This commit introduces functionality to query the workspace size
as required by a tir primfunc by looking at tir.allocates inside of it
Change-Id: I6f8ca90408b6e35d17ec818998a0f158a268a2a6
* Add workspace size calculation for primfuncs
*change int --> size_t
Change-Id: If7fafec0269937d70184e7696e44386b74116d86
* Add workspace size calculation for primfuncs
* int --> size_t change for analysis.h
Change-Id: I9e5c5e5f8458663390c50cf56f1a11910687928d
* Add workspace size calculation for primfuncs
* lambda scope fix
Change-Id: I0c9b4c529150de8e0a5e170887cc935b7d0f6af2
Co-authored-by: Chenfan <jcf94@outlook.com>
[TIR][REFACTOR] ForNode update
- Remove deprecated device_api.
- Add ThreadBinding for_type.
- Add additional annotations.
More style consistency refactor to make the ForNode
to be consistent with rest of the codebase.
- ForType => ForKind
- Add constant prefix k to enum consts per Google C style
- Introduce ForKind to the python side.
* [TIR][REFACTOR] Enforce allocate to only accept buffer_var with correct PtrType.
This is a refactoring step to cleanup legacy issue of opaque buffer
var without ptr type information. Now all the allocation comes with the right
pointer data type. Places touched:
- TVMScript Parser: add the right info to get the correct pointer type.
- Cross thread all reduce: set the right pointer type.
- Storage rewrite: setup the right pointer type.
- Custom dtype: remap the variables with new pointer type.
x
* Address comments
Co-authored-by: Tristan Konolige <tristan.konolige@gmail.com>
Co-authored-by: Tristan Konolige <tristan.konolige@gmail.com>
Siyuan Feng