| 20 | namespace oneapi { |
| 21 | template<af_op_t op, typename Ti, typename Tk, typename To> |
| 22 | Array<To> scan(const Array<Tk>& key, const Array<Ti>& in, const int dim, |
| 23 | bool inclusive_scan) { |
| 24 | ONEAPI_NOT_SUPPORTED("scan Not supported"); |
| 25 | |
| 26 | Array<To> out = createEmptyArray<To>(in.dims()); |
| 27 | |
| 28 | // Param Out = out; |
| 29 | // Param Key = key; |
| 30 | // Param In = in; |
| 31 | |
| 32 | // if (dim == 0) { |
| 33 | // // kernel::scanFirstByKey<Ti, Tk, To, op>(Out, In, Key, |
| 34 | // inclusive_scan); |
| 35 | // } else { |
| 36 | // // kernel::scanDimByKey<Ti, Tk, To, op>(Out, In, Key, dim, |
| 37 | // inclusive_scan); |
| 38 | // } |
| 39 | return out; |
| 40 | } |
| 41 | |
| 42 | #define INSTANTIATE_SCAN_BY_KEY(ROp, Ti, Tk, To) \ |
| 43 | template Array<To> scan<ROp, Ti, Tk, To>( \ |