Fix docs [skip ci]

roflmaostc · Dec 4, 2023 · 1929a6b · 1929a6b
1 parent 0c33fb6
commit 1929a6b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/docs/src/benchmark.md b/docs/src/benchmark.md
@@ -33,6 +33,19 @@ simshow(sinogram)
 
 CUDA.@time CUDA.@sync sinogram_c = radon(sample_2D_c, angles_c, backend=CUDABackend());
 CUDA.@time CUDA.@sync sample_iradon_c = iradon(sinogram_c, angles_c, backend=CUDABackend());
+
+
+sample_3D = randn(Float32, (512, 512, 100));
+sample_3D_c = CuArray(sample_3D)
+
+@time sinogram_3D = radon(sample_3D, angles);
+@benchmark radon($sample_3D, $angles)
+@benchmark iradon($sinogram_3D, $angles)
+
+sinogram_3D_c = radon(sample_3D_c, angles_c, backend=CUDABackend())
+@benchmark CUDA.@sync radon($sample_3D_c, $angles_c, backend=CUDABackend())
+@benchmark CUDA.@sync iradon($sinogram_3D_c, $angles_c, backend=CUDABackend())
+
 ```
 ![](../assets/radonka_sinogram.png)
 ![](../assets/radonka_iradon.png)