CUDA работает!!

bitcoin-project/src/gpu_loader.cpp

@@ -0,0 +1,12 @@
+#include "gpu_loader.hpp"
+#include <dlfcn.h>
+
+gpu_is_available_fn load_gpu_is_available() {
+    void* h = dlopen("./libgpu_compute.so", RTLD_NOW | RTLD_LOCAL);
+    if (!h) return nullptr;
+
+    auto fn = (gpu_is_available_fn)dlsym(h, "gpu_is_available");
+    if (!fn) return nullptr;
+
+    return fn;
+}

bitcoin-project/src/gpu_loader.hpp

@@ -0,0 +1,4 @@
+#pragma once
+using gpu_is_available_fn = int (*)();
+
+gpu_is_available_fn load_gpu_is_available();

bitcoin-project/src/gpu_plugin.cu

@@ -0,0 +1,8 @@
+#include <cuda_runtime.h>
+
+extern "C" int gpu_is_available() {
+    int n = 0;
+    cudaError_t err = cudaGetDeviceCount(&n);
+    if (err != cudaSuccess) return 0;
+    return (n > 0) ? 1 : 0;
+}

bitcoin-project/src/main.cpp

@@ -6,6 +6,7 @@
 #include "csv_loader.hpp"
 #include "utils.hpp"
 #include "record.hpp"
+#include "gpu_loader.hpp"
 
 // Функция: отобрать записи для конкретного ранга
 std::vector<Record> select_records_for_rank(
@@ -35,7 +36,7 @@ int main(int argc, char** argv) {
     if (rank == 0) {
         std::cout << "Rank 0 loading CSV..." << std::endl;
 
-        auto records = load_csv("data/data.csv");
+        auto records = load_csv("/mnt/shared/supercomputers/bitcoin-project/data/data.csv");
 
         auto days = group_by_day(records);
         auto parts = split_days(days, size);
@@ -70,6 +71,16 @@ int main(int argc, char** argv) {
     std::cout << "Rank " << rank << " received "
               << local_records.size() << " records" << std::endl;
 
+
+    auto gpu_is_available = load_gpu_is_available();
+    int have_gpu = 0;
+    if (gpu_is_available) {
+        std::cout << "Rank " << rank << " dll loaded" << std::endl;
+        have_gpu = gpu_is_available();
+    }
+
+    std::cout << "Rank " << rank << ": gpu_available=" << have_gpu << "\n";
+
     MPI_Finalize();
     return 0;
 }