Skip to content

Commit 79d4f59

Browse files
committed
Cufile doc claims O_DIRECT is not necessary anymore
1 parent f6a0a6b commit 79d4f59

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

cufile/cufile.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,21 +31,23 @@ def _os_mode(mode: str):
3131
"a": os.O_CREAT | os.O_WRONLY | os.O_APPEND,
3232
"a+": os.O_CREAT | os.O_RDWR | os.O_APPEND,
3333
}
34-
return modes[mode] | os.O_DIRECT
34+
return modes[mode]
3535

3636
class CuFile:
3737
"""
3838
Main class for CUDA file operations.
3939
"""
4040

41-
def __init__(self, path: str, mode: str = "r"):
41+
def __init__(self, path: str, mode: str = "r", use_direct_io: bool = False):
4242
"""
4343
Initialize the CuFile instance.
4444
"""
4545
self._driver = CuFileDriver()
4646
self._path = path
4747
self._mode = mode
4848
self._os_mode = _os_mode(mode)
49+
if use_direct_io:
50+
self._os_mode |= os.O_DIRECT
4951

5052
def __enter__(self):
5153
"""

0 commit comments

Comments
 (0)