CUDA device API & VerifyGPUCode pass update (#5898)
authorChenfan <jcf94@outlook.com>
Thu, 25 Jun 2020 05:44:39 +0000 (13:44 +0800)
committerGitHub <noreply@github.com>
Thu, 25 Jun 2020 05:44:39 +0000 (22:44 -0700)
commit074a07ede031762afb755f8b9273c0afb55c8b4a
tree0cdbcbb05aab2f123903a7f99ae60205996efd0c
parent524552a81123540e4187bac735d6c54eb6e5c662
CUDA device API & VerifyGPUCode pass update (#5898)

* Add kMaxRegistersPerBlock device api for cuda

* Add vectorize check to verify_gpu_code

* Lint fix

* Cast fix
include/tvm/runtime/device_api.h
src/runtime/cuda/cuda_device_api.cc
src/runtime/metal/metal_device_api.mm
src/runtime/opencl/opencl_device_api.cc
src/runtime/rocm/rocm_device_api.cc
src/runtime/vulkan/vulkan.cc
src/tir/analysis/verify_gpu_code.cc
tests/python/unittest/test_tir_analysis_verify_gpu_code.py