GdsFile

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 的包裝器。將從檔案中的 offset 位置載入 storage.nbytes() 位元組的資料到儲存中。

引數

• storage (Storage) – 要載入資料的儲存。

• offset (int, optional) – 檔案中的偏移量，從該位置開始載入。 (預設: 0)

register_handle()

向 cuFile 驅動程式註冊檔案描述符。

這是 cuFileHandleRegister 的包裝器。

save_storage(storage, offset=0)

將儲存中的資料儲存到檔案中。

這是 cuFileWrite 的包裝器。儲存中的所有位元組將寫入檔案中的 offset 位置。

引數

• storage (Storage) – 要從中儲存資料的儲存。

• offset (int, optional) – 檔案中的偏移量，從該位置開始儲存。 (預設: 0)