| 64 | } |
| 65 | |
| 66 | void padding_to_workspace(_megdnn_tensor_out dst, _megdnn_tensor_in src) { |
| 67 | switch (src.layout.dtype.enumv()) { |
| 68 | #define cb(name, ctype) \ |
| 69 | case (DTypeEnum::name): { \ |
| 70 | recursive_cp<ctype>(dst, src); \ |
| 71 | break; \ |
| 72 | } |
| 73 | |
| 74 | cb(Float32, dt_float32); |
| 75 | cb(Int32, dt_int32); |
| 76 | cb(QuantizedS32, dt_int32); |
| 77 | cb(QuantizedS8, dt_qint8); |
| 78 | #undef cb |
| 79 | #define cb(name, size_nbits) \ |
| 80 | case (DTypeEnum::name): { \ |
| 81 | lowbit_recursive_cp<size_nbits>(dst, src); \ |
| 82 | break; \ |
| 83 | } |
| 84 | cb(QuantizedS4, 4); |
| 85 | cb(Quantized4Asymm, 4); |
| 86 | #undef cb |
| 87 | default: |
| 88 | megdnn_assert(0, "not support dtype %s", src.layout.dtype.name()); |
| 89 | } |
| 90 | } |
| 91 | |
| 92 | void extract_from_workspace( |
| 93 | _megdnn_tensor_out dst, _megdnn_tensor_in src, size_t group) { |
no test coverage detected