Coverage for src / cufile_patcher / __init__.py: 100%

1"""cufile-patcher: plugin-oriented cuFile wrapper and extension scaffold.""" 

2 

3from .auto_patch import AutoPatchContext, auto_patch 

4from .bindings import ( 

5 cuFileDriverClose, 

6 cuFileDriverOpen, 

7 cuFileHandleDeregister, 

8 cuFileHandleRegister, 

9 cuFileRead, 

10 cuFileWrite, 

11 get_default_backend, 

12 register_backend, 

13 set_default_backend, 

14) 

15from .core import hello_world 

16from .cufile import CuFile, CuFileDriver 

17from .safetensor_patcher import ( 

18 CuFileSafeTensorStreamReader, 

19 PythonSafeTensorStreamReader, 

20 SafeTensorCuFilePatcher, 

21 SafeTensorStreamReader, 

22 patch_safetensor_load_file, 

23) 

24from .tensorflow_patcher import ( 

25 CuFileTFStreamReader, 

26 PythonTFStreamReader, 

27 TensorFlowCuFilePatcher, 

28 TensorFlowStreamReader, 

29 patch_tensorflow_load_model, 

30) 

31from .torch_patcher import ( 

32 CuFileStreamReader, 

33 PythonStreamReader, 

34 TorchCuFilePatcher, 

35 TorchStreamReader, 

36 patch_torch_load, 

37) 

38 

39__all__ = [ 

40 "CuFile", 

41 "CuFileDriver", 

42 "AutoPatchContext", 

43 "auto_patch", 

44 "cuFileDriverClose", 

45 "cuFileDriverOpen", 

46 "cuFileHandleDeregister", 

47 "cuFileHandleRegister", 

48 "cuFileRead", 

49 "cuFileWrite", 

50 "get_default_backend", 

51 "hello_world", 

52 "register_backend", 

53 "set_default_backend", 

54 "PythonStreamReader", 

55 "CuFileStreamReader", 

56 "TorchCuFilePatcher", 

57 "TorchStreamReader", 

58 "patch_torch_load", 

59 "CuFileSafeTensorStreamReader", 

60 "PythonSafeTensorStreamReader", 

61 "SafeTensorCuFilePatcher", 

62 "SafeTensorStreamReader", 

63 "patch_safetensor_load_file", 

64 "CuFileTFStreamReader", 

65 "PythonTFStreamReader", 

66 "TensorFlowCuFilePatcher", 

67 "TensorFlowStreamReader", 

68 "patch_tensorflow_load_model", 

69]