metal-bandit/test_gpu_optimization.jl at main · tatsujit/metal-bandit · GitHub

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
using Metal
using Statistics
using Random

# Include our optimization
include("gpu_optimized_simple.jl")

# Test the optimization
println("🚀 Testing GPU Optimization...")
result = test_gpu_acceleration()

# Display results
println("\n🏆 GPU Optimization Results:")
println("=" ^ 40)
gpu_time = result.gpu_result.total_time
original_time = result.original_time
speedup = result.speedup

println("GPU Time: $(round(gpu_time, digits=2))s")
if original_time !== nothing
    println("Original Time: $(round(original_time, digits=2))s")
    println("Speedup: $(round(speedup, digits=2))x")
    if speedup > 1.0
        println("✅ GPU IS FASTER!")
    else
        println("❌ GPU needs more optimization")
    end
else
    println("Original comparison not available")
end

println("\n📊 GPU Acceleration Summary:")
println("α correlation: $(round(result.gpu_result.alpha_correlation, digits=3))")
println("β correlation: $(round(result.gpu_result.beta_correlation, digits=3))")