zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Remove duplicate defines of StreamOrDevice and is_big_endian (#892)

Browse Source
This commit is contained in:
Cheng
2024-03-27 07:15:11 +09:00
committed by GitHub
parent 240d10699c
commit a789685c63
4 changed files with 14 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mlx/utils.h
View File

@@ -84,6 +84,16 @@ int check_shape_dim(const T dim) {
return static_cast<int>(dim);
}
inline bool is_big_endian() {
union ByteOrder {
int32_t i;
uint8_t c[4];
};
ByteOrder b = {0x01234567};
return b.c[0] == 0x01;
}
/**
* Returns the axis normalized to be in the range [0, ndim).
* Based on numpy's normalize_axis_index. See