| 49 | |
| 50 | template<typename To, typename Ti> |
| 51 | void median_flat(int nx, int ny = 1, int nz = 1, int nw = 1) { |
| 52 | SUPPORTED_TYPE_CHECK(Ti); |
| 53 | array a = generateArray<Ti>(nx, ny, nz, nw); |
| 54 | |
| 55 | // Verification |
| 56 | array sa = sort(flat(a)); |
| 57 | dim_t mid = (sa.dims(0) + 1) / 2; |
| 58 | |
| 59 | To verify; |
| 60 | |
| 61 | To *h_sa = sa.as((af_dtype)dtype_traits<To>::af_type).host<To>(); |
| 62 | if (sa.dims(0) % 2 == 1) { |
| 63 | verify = h_sa[mid - 1]; |
| 64 | } else { |
| 65 | verify = (h_sa[mid - 1] + h_sa[mid]) / (To)2; |
| 66 | } |
| 67 | |
| 68 | // Test Part |
| 69 | To val = median<To>(a); |
| 70 | |
| 71 | ASSERT_EQ(verify, val); |
| 72 | |
| 73 | af_free_host(h_sa); |
| 74 | } |
| 75 | |
| 76 | template<typename To, typename Ti, int dim> |
| 77 | void median_test(int nx, int ny = 1, int nz = 1, int nw = 1) { |
nothing calls this directly
no test coverage detected