Чтение данных видеокартой с диска nvme ssd

#include <iostream>
#include <cuda_runtime.h>
#include <cufile.h>
#include <fcntl.h>
#include <unistd.h>
#include <cstring>
 
int main() {
    constexpr size_t SIZE = 4096; // 4 KB
 
    // 1. Инициализация GPUDirect Storage
    if (auto status = cuFileDriverOpen(); status.err != CU_FILE_SUCCESS) {
        std::cerr << "cuFileDriverOpen failed: " 
                  << status.err << std::endl;
        return 1;
    }
 
    // 2. Открываем файл (можно /dev/nvme0n1 для raw чтения)
    const char* filename = "/mnt/ssd/test.bin";
    int fd = open(filename, O_RDONLY | O_DIRECT);
    if (fd < 0) {
        perror("open");
        return 1;
    }
 
    // 3. Регистрируем дескриптор файла в cuFile
    CUfileDescr_t desc{};
    desc.handle.fd = fd;
    desc.type = CU_FILE_HANDLE_TYPE_OPAQUE_FD;
 
    CUfileHandle_t cfHandle;
    if (auto status = cuFileHandleRegister(&cfHandle, &desc); status.err != CU_FILE_SUCCESS) {
        std::cerr << "cuFileHandleRegister failed: " 
                  << status.err << std::endl;
        close(fd);
        return 1;
    }
 
    // 4. Выделяем память на GPU
    void* d_buf = nullptr;
    if (cudaMalloc(&d_buf, SIZE) != cudaSuccess) {
        std::cerr << "cudaMalloc failed" << std::endl;
        cuFileHandleDeregister(cfHandle);
        close(fd);
        return 1;
    }
 
    // Регистрируем GPU-буфер для GDS
    if (auto status = cuFileBufRegister(d_buf, SIZE, 0); status.err != CU_FILE_SUCCESS) {
        std::cerr << "cuFileBufRegister failed: " 
                  << status.err << std::endl;
        cudaFree(d_buf);
        cuFileHandleDeregister(cfHandle);
        close(fd);
        return 1;
    }
 
    // 5. Читаем файл в GPU память
    ssize_t ret = cuFileRead(cfHandle, d_buf, SIZE, 0, 0);
    if (ret < 0) {
        std::cerr << "cuFileRead failed" << std::endl;
    } else {
        std::cout << "Прочитано " << ret << " байт напрямую в GPU память" << std::endl;
    }
 
    // 6. Очистка
    cuFileBufDeregister(d_buf);
    cudaFree(d_buf);
    cuFileHandleDeregister(cfHandle);
    close(fd);
    cuFileDriverClose();
 
    return 0;
}