-
- Downloads
[CODEGEN] fix vector conversion for opencl (#783)
* support more argument type in depthwise_conv2d * mark all pointer as 'restrict' & fix vector conversion for opencl
Showing
- src/codegen/codegen_c.cc 11 additions, 11 deletionssrc/codegen/codegen_c.cc
- src/codegen/codegen_c.h 2 additions, 2 deletionssrc/codegen/codegen_c.h
- src/codegen/codegen_opencl.cc 16 additions, 0 deletionssrc/codegen/codegen_opencl.cc
- src/codegen/codegen_opencl.h 2 additions, 0 deletionssrc/codegen/codegen_opencl.h
- src/pass/split_host_device.cc 0 additions, 3 deletionssrc/pass/split_host_device.cc
- topi/python/topi/nn/depthwise_conv2d.py 18 additions, 4 deletionstopi/python/topi/nn/depthwise_conv2d.py
- topi/python/topi/testing/depthwise_conv2d_python.py 10 additions, 2 deletionstopi/python/topi/testing/depthwise_conv2d_python.py
- topi/tests/python/test_topi_depthwise_conv2d.py 3 additions, 4 deletionstopi/tests/python/test_topi_depthwise_conv2d.py
Loading
Please register or sign in to comment