We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f6a0a6b commit 79d4f59Copy full SHA for 79d4f59
cufile/cufile.py
@@ -31,21 +31,23 @@ def _os_mode(mode: str):
31
"a": os.O_CREAT | os.O_WRONLY | os.O_APPEND,
32
"a+": os.O_CREAT | os.O_RDWR | os.O_APPEND,
33
}
34
- return modes[mode] | os.O_DIRECT
+ return modes[mode]
35
36
class CuFile:
37
"""
38
Main class for CUDA file operations.
39
40
41
- def __init__(self, path: str, mode: str = "r"):
+ def __init__(self, path: str, mode: str = "r", use_direct_io: bool = False):
42
43
Initialize the CuFile instance.
44
45
self._driver = CuFileDriver()
46
self._path = path
47
self._mode = mode
48
self._os_mode = _os_mode(mode)
49
+ if use_direct_io:
50
+ self._os_mode |= os.O_DIRECT
51
52
def __enter__(self):
53
0 commit comments