[TIR] Make lower_warp_memory support extent(threadIdx.x) < warp_size (#5307)
authorTang, Shizhi <rd0x01@gmail.com>
Fri, 17 Apr 2020 14:35:11 +0000 (22:35 +0800)
committerGitHub <noreply@github.com>
Fri, 17 Apr 2020 14:35:11 +0000 (07:35 -0700)
commit4b5f324ad17571d2fbcef44d719dae7772b66e91
treeb1033dd54550769dd3e4c891a8ec7faf178d9fc1
parent49d304fcfa29f9ef99be2761187347cf8ca40ea8
[TIR] Make lower_warp_memory support extent(threadIdx.x) < warp_size (#5307)

* support extent(threadIdx.x) < warp_size in lower_warp_memory

* more docs for lower_warp_memory
include/tvm/tir/expr.h
src/target/source/intrin_rule_cuda.cc
src/target/source/intrin_rule_opencl.cc
src/tir/transforms/lower_warp_memory.cc
tests/python/unittest/test_tir_transform_lower_warp_memory.py