decuda 0.1.1

CUDA to HIP, SYCL, OpenCL, and Rust GPU migration tool — automatic source-code translator for porting CUDA C++ kernels to AMD ROCm HIP, Intel oneAPI SYCL, Khronos OpenCL, and Rust GPU (cust / rust-gpu)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

// Generated by decuda.
// Host-side runtime calls are translated to the `cust` crate. Kernels
// are emitted as `TODO(decuda)` blocks: rust-gpu translation requires
// the kernel to be authored as a Rust fn. See `examples/` for a
// scaffolded SPIR-V kernel module you can flesh out.
//
// Add to your Cargo.toml:
//   [dependencies]
//   cust = "0.3"

// Advanced fixture: 3D 7-point stencil with shared-memory halo exchange.
//
// Exercises:
//   - 3D grid and block (dim3 with three components)
//   - threadIdx.{x,y,z}, blockIdx.{x,y,z}, blockDim.{x,y,z}, gridDim.{x,y,z}
//   - __shared__ 3D tile with halo
//   - __syncthreads barriers
//   - __global__ kernel with multi-dimensional indexing
//   - cudaMalloc, cudaFree, cudaMemcpy
//   - cuda_runtime.h header
// was: #include cuda_runtime.h  ->  cust::cuda_build_setup() /* TODO: import cust crate */

#define BX 8
#define BY 8
#define BZ 8
#define HALO 1

// TODO(decuda): rewrite as rust-gpu kernel fn
 void stencil_3d(const float* in, float* out, int nx, int ny, int nz) {
    // Shared memory tile: interior + 1-cell halo on each side.
    /* shared -> rust-gpu group_memory */ float tile[BZ + 2 * HALO][BY + 2 * HALO][BX + 2 * HALO];

    int tx = thread_idx;
    int ty = thread_idx;
    int tz = thread_idx;

    int gx = block_idx * block_dim + tx;
    int gy = block_idx * block_dim + ty;
    int gz = block_idx * block_dim + tz;

    int idx = gx + gy * nx + gz * nx * ny;

    // Load interior.
    if (gx < nx && gy < ny && gz < nz) {
        tile[tz + HALO][ty + HALO][tx + HALO] = in[idx];
    }

    // Load halos — each thread loads one halo cell from each face.
    // X-minus halo.
    if (tx < HALO && gx >= HALO && gy < ny && gz < nz) {
        tile[tz + HALO][ty + HALO][tx] = in[idx - HALO];
    }
    // X-plus halo.
    if (tx >= block_dim - HALO && gx + HALO < nx && gy < ny && gz < nz) {
        tile[tz + HALO][ty + HALO][tx + 2 * HALO] = in[idx + HALO];
    }
    // Y-minus halo.
    if (ty < HALO && gy >= HALO && gx < nx && gz < nz) {
        tile[tz + HALO][ty][tx + HALO] = in[idx - HALO * nx];
    }
    // Y-plus halo.
    if (ty >= block_dim - HALO && gy + HALO < ny && gx < nx && gz < nz) {
        tile[tz + HALO][ty + 2 * HALO][tx + HALO] = in[idx + HALO * nx];
    }
    // Z-minus halo.
    if (tz < HALO && gz >= HALO && gx < nx && gy < ny) {
        tile[tz][ty + HALO][tx + HALO] = in[idx - HALO * nx * ny];
    }
    // Z-plus halo.
    if (tz >= block_dim - HALO && gz + HALO < nz && gx < nx && gy < ny) {
        tile[tz + 2 * HALO][ty + HALO][tx + HALO] = in[idx + HALO * nx * ny];
    }

    group.sync();

    // 7-point stencil computation.
    if (gx < nx && gy < ny && gz < nz) {
        float c = tile[tz + HALO][ty + HALO][tx + HALO];
        float xp = tile[tz + HALO][ty + HALO][tx + HALO + 1];
        float xm = tile[tz + HALO][ty + HALO][tx + HALO - 1];
        float yp = tile[tz + HALO][ty + HALO + 1][tx + HALO];
        float ym = tile[tz + HALO][ty + HALO - 1][tx + HALO];
        float zp = tile[tz + HALO + 1][ty + HALO][tx + HALO];
        float zm = tile[tz + HALO - 1][ty + HALO][tx + HALO];
        out[idx] = 0.125f * (xp + xm + yp + ym + zp + zm) + c;
    }
}

int main(void) {
    const int NX = 64;
    const int NY = 64;
    const int NZ = 64;
    float* d_in = nullptr;
    float* d_out = nullptr;

    cudaMalloc((void**)&d_in, NX * NY * NZ * sizeof(float));
    cudaMalloc((void**)&d_out, NX * NY * NZ * sizeof(float));
    cudaMemcpy(d_in, d_in, NX * NY * NZ * sizeof(float), cudaMemcpyDeviceToDevice);

    dim3 grid(NX / BX, NY / BY, NZ / BZ);
    dim3 block(BX, BY, BZ);
    { /* decuda cust launch */ let _kernel = modules.get_function("stencil_3d"); unsafe { let _ = launch!( _kernel<<<grid as grid_size, block as block_size, 0 as usize, default>>>(d_in, d_out, NX, NY, NZ) ); } };

    cudaDeviceSynchronize();
    cudaFree(d_in);
    cudaFree(d_out);
    return 0;
}