Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix MultilayerQG module on GPU & Julia v1.6 #238

Merged
merged 1 commit into from
May 20, 2021

Conversation

navidcy
Copy link
Member

@navidcy navidcy commented May 20, 2021

No description provided.

@@ -202,7 +202,7 @@ function Params(nlayers, g, f₀, β, ρ, H, U, eta, μ, ν, nν, grid; calcFq=n
U = convert_U_to_U3D(dev, nlayers, grid, U)

Uyy = real.(ifft(-l.^2 .* fft(U)))
Uyy = repeat(Uyy, outer=(nx, 1, 1))
Uyy = CUDA.@allowscalar repeat(Uyy, outer=(nx, 1, 1))
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why repeat? Would reshape be enough?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

real.(ifft(-l.^2 .* fft(U))) returns an (1, ny, nlayers) array and what we need is (nx, ny, nlayers)...

@navidcy navidcy merged commit b05af84 into master May 20, 2021
@navidcy navidcy deleted the ncc/multilayerqg-gpu-on-julia-v1.6 branch May 20, 2021 01:41
@navidcy navidcy linked an issue May 27, 2021 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

MultiLayerQG tests fail on Julia v1.6
2 participants