Skip to content

Commit 0beecb4

Browse files
kuarorafacebook-github-bot
authored andcommitted
sys.big_endian to sys.byteorder (facebookresearch#3422)
Summary: Pull Request resolved: facebookresearch#3422 Found vec_io failing when running some benchmarking. There is no such field named big_endian in sys. So, reverting it to original field byteorder Reviewed By: algoriddle Differential Revision: D56718607 fbshipit-source-id: 553f1d2d6bc967581142a92282e534f3f164e8f9
1 parent 6e7d9e0 commit 0beecb4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

contrib/vecs_io.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
def ivecs_read(fname):
1616
a = np.fromfile(fname, dtype='int32')
17-
if sys.big_endian:
17+
if sys.byteorder == 'big':
1818
a.byteswap(inplace=True)
1919
d = a[0]
2020
return a.reshape(-1, d + 1)[:, 1:].copy()
@@ -25,7 +25,7 @@ def fvecs_read(fname):
2525

2626

2727
def ivecs_mmap(fname):
28-
assert not sys.big_endian
28+
assert sys.byteorder != 'big'
2929
a = np.memmap(fname, dtype='int32', mode='r')
3030
d = a[0]
3131
return a.reshape(-1, d + 1)[:, 1:]
@@ -37,7 +37,7 @@ def fvecs_mmap(fname):
3737

3838
def bvecs_mmap(fname):
3939
x = np.memmap(fname, dtype='uint8', mode='r')
40-
if sys.big_endian:
40+
if sys.byteorder == 'big':
4141
da = x[:4][::-1].copy()
4242
d = da.view('int32')[0]
4343
else:
@@ -50,7 +50,7 @@ def ivecs_write(fname, m):
5050
m1 = np.empty((n, d + 1), dtype='int32')
5151
m1[:, 0] = d
5252
m1[:, 1:] = m
53-
if sys.big_endian:
53+
if sys.byteorder == 'big':
5454
m1.byteswap(inplace=True)
5555
m1.tofile(fname)
5656

0 commit comments

Comments
 (0)