Gds 文件 ¶

class torch.cuda.gds.GdsFile(filename, flags)

cuFile 的包装器。

cuFile 是一个对 GPUDirect Storage (GDS) API 的文件接口。

请参阅 cufile 文档以获取更多详细信息。

参数:

• filename (str) – 要打开的文件名。

• flags (int) – 打开文件时传递给 os.open 的标志。 os.O_DIRECT 将自动添加。

示例：

>>> src1 = torch.randn(1024, device="cuda")
>>> src2 = torch.randn(2, 1024, device="cuda")
>>> file = torch.cuda.gds.GdsFile(f, os.O_CREAT | os.O_RDWR)
>>> file.save_storage(src1.untyped_storage(), offset=0)
>>> file.save_storage(src2.untyped_storage(), offset=src1.nbytes)
>>> dest1 = torch.empty(1024, device="cuda")
>>> dest2 = torch.empty(2, 1024, device="cuda")
>>> file.load_storage(dest1.untyped_storage(), offset=0)
>>> file.load_storage(dest2.untyped_storage(), offset=src1.nbytes)
>>> torch.equal(src1, dest1)
True
>>> torch.equal(src2, dest2)
True

deregister_handle()

从 cuFile 驱动程序中注销文件描述符。

这是围绕 cuFileHandleDeregister 的包装器。

load_storage(storage, offset=0)

从文件中加载数据到存储中。

这是一个围绕 cuFileRead 的包装器。 storage.nbytes() 的数据将从文件 offset 加载到存储中。

参数:

• 存储（Storage）- 将数据加载到其中的存储。

• 偏移量（int，可选）- 从文件中开始加载的偏移量。（默认：0）

register_handle()

将文件描述符注册到 cuFile 驱动程序。

这是围绕 cuFileHandleRegister 的包装器。

save_storage(storage, offset=0)

将存储数据保存到文件中。

这是围绕 cuFileWrite 的包装器。所有存储的字节将被写入到文件 offset 中。

参数:

• 存储（Storage）- 用于保存数据的存储。

• 偏移量（int，可选）- 从文件开始保存的偏移量。（默认：0）