| 90 | : GroupLocalForward(handle) {} |
| 91 | |
| 92 | size_t GroupLocalForwardImpl::get_workspace_in_bytes( |
| 93 | const TensorLayout& src, const TensorLayout& filter, const TensorLayout& dst) { |
| 94 | if (prefer_inference_kernel(src, filter, dst)) { |
| 95 | return 0; |
| 96 | } else { |
| 97 | auto G = filter[0]; |
| 98 | TensorLayout src_g = prepare_src_dst(src, G); |
| 99 | TensorLayout dst_g = prepare_src_dst(dst, G); |
| 100 | TensorLayout filter_g = prepare_filter(filter); |
| 101 | return get_opr(handle(), param()) |
| 102 | ->get_workspace_in_bytes(src_g, filter_g, dst_g); |
| 103 | } |
| 104 | } |
| 105 | |
| 106 | bool GroupLocalForwardImpl::prefer_inference_kernel( |
| 107 | const TensorLayout& src, const TensorLayout& filter, const TensorLayout& dst) { |
nothing calls this directly
no test coverage detected