[TOPI] conv2d nchw gpu scheduler (#315)
* __init__ updated * pull request updated * build_module added * typo fixed * another typo fixed * conv2d gpu scheduler for two layouts moved to tvm * changes made according to CR * conv2d_nchw formating updated, conv2d_hwcn tests updated * lint error fixed * element wise operator schedule fusing fixed for conv2d * conv2d_nchw topi test added, all resnet workloads now pass * conv compute lint error fixed * fixed python 3 compatibility problem * conv2d tensor input support added, test typo fixed, ir_pass.Simplify changed to util.get_const_int
Showing
- topi/python/topi/cuda/__init__.py 2 additions, 1 deletiontopi/python/topi/cuda/__init__.py
- topi/python/topi/cuda/conv2d_hwcn.py 119 additions, 0 deletionstopi/python/topi/cuda/conv2d_hwcn.py
- topi/python/topi/cuda/conv2d_hwcn_map.py 0 additions, 121 deletionstopi/python/topi/cuda/conv2d_hwcn_map.py
- topi/python/topi/cuda/conv2d_nchw.py 137 additions, 0 deletionstopi/python/topi/cuda/conv2d_nchw.py
- topi/python/topi/nn/conv.py 64 additions, 1 deletiontopi/python/topi/nn/conv.py
- topi/python/topi/testing/__init__.py 1 addition, 0 deletionstopi/python/topi/testing/__init__.py
- topi/python/topi/testing/conv2d_hwcn_python.py 1 addition, 1 deletiontopi/python/topi/testing/conv2d_hwcn_python.py
- topi/python/topi/testing/conv2d_nchw_python.py 64 additions, 0 deletionstopi/python/topi/testing/conv2d_nchw_python.py
- topi/recipe/conv/test_conv2d_hwcn_map.py 3 additions, 3 deletionstopi/recipe/conv/test_conv2d_hwcn_map.py
- topi/tests/python/test_topi_conv2d_hwcn_map.py 2 additions, 2 deletionstopi/tests/python/test_topi_conv2d_hwcn_map.py
- topi/tests/python/test_topi_conv2d_nchw.py 61 additions, 0 deletionstopi/tests/python/test_topi_conv2d_nchw.py
topi/python/topi/cuda/conv2d_hwcn.py
0 → 100644
topi/python/topi/cuda/conv2d_nchw.py
0 → 100644
topi/tests/python/test_topi_conv2d_nchw.py
0 → 100644
Please register or sign in to comment