| 231 | } |
| 232 | |
| 233 | af_err af_get_allocated_bytes(size_t *bytes, const af_array arr) { |
| 234 | try { |
| 235 | af_dtype ty = getInfo(arr).getType(); |
| 236 | |
| 237 | size_t res = 0; |
| 238 | |
| 239 | switch (ty) { |
| 240 | case f32: res = getArray<float>(arr).getAllocatedBytes(); break; |
| 241 | case f64: res = getArray<double>(arr).getAllocatedBytes(); break; |
| 242 | case c32: res = getArray<cfloat>(arr).getAllocatedBytes(); break; |
| 243 | case c64: res = getArray<cdouble>(arr).getAllocatedBytes(); break; |
| 244 | case u32: res = getArray<uint>(arr).getAllocatedBytes(); break; |
| 245 | case s32: res = getArray<int>(arr).getAllocatedBytes(); break; |
| 246 | case u64: res = getArray<uintl>(arr).getAllocatedBytes(); break; |
| 247 | case s64: res = getArray<intl>(arr).getAllocatedBytes(); break; |
| 248 | case u16: res = getArray<ushort>(arr).getAllocatedBytes(); break; |
| 249 | case s16: res = getArray<short>(arr).getAllocatedBytes(); break; |
| 250 | case b8: res = getArray<char>(arr).getAllocatedBytes(); break; |
| 251 | case u8: res = getArray<uchar>(arr).getAllocatedBytes(); break; |
| 252 | case s8: res = getArray<schar>(arr).getAllocatedBytes(); break; |
| 253 | case f16: res = getArray<half>(arr).getAllocatedBytes(); break; |
| 254 | default: TYPE_ERROR(6, ty); |
| 255 | } |
| 256 | |
| 257 | std::swap(*bytes, res); |
| 258 | } |
| 259 | CATCHALL; |
| 260 | return AF_SUCCESS; |
| 261 | } |
no test coverage detected