| 10 | |
| 11 | namespace { |
| 12 | void bang_c_wrapper( |
| 13 | uint32_t* dst, const uint32_t* src, int nr_elems, cnrtQueue_t queue, |
| 14 | int core_version) { |
| 15 | checksum_kernel_union1_wrapper(dst, src, nr_elems, queue); |
| 16 | // if (core_version == 220) { |
| 17 | // checksum_kernel_union1_wrapper(dst, src, nr_elems, queue); |
| 18 | // } else { |
| 19 | // checksum_kernel_union4_wrapper(dst, src, nr_elems, queue); |
| 20 | // } |
| 21 | after_kernel_launch(); |
| 22 | } |
| 23 | } // namespace |
| 24 | |
| 25 | size_t ChecksumForwardImpl::get_workspace_in_bytes(const TensorLayout& /* data */) { |