Skip to content

Commit

Permalink
fix type for other GPUs;
Browse files Browse the repository at this point in the history
correctly uses Float32
  • Loading branch information
korbinian90 committed Oct 10, 2023
1 parent 34b7e05 commit 922d66e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/tgv.jl
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ function set_parameters(alpha, res, B0_dir, cu)

resinv = cu(1 ./ res)
laplace_kernel = cu(res .^ -2)
dipole_kernel = cu((1 / 3 .- B0_dir .^ 2) ./ (res .^ 2))
dipole_kernel = cu((1 // 3 .- B0_dir .^ 2) ./ (res .^ 2))

return alphainv, tau, sigma, resinv, laplace_kernel, dipole_kernel
end
Expand Down

0 comments on commit 922d66e

Please sign in to comment.