Skip to content

field-line tracing for axisymmetric EQDSK equilibria #408

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.

Sign up for GitHub

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

Draft
abhiyanpaudel wants to merge 13 commits into
base: develop
Choose a base branch
from
Draft

field-line tracing for axisymmetric EQDSK equilibria #408

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
6d89bcc
Add axisymmetric magnetic field from EQDSK
abhiyan123 Oct 29, 2026
166e2bf
Add batched field-line tracer
abhiyan123 Oct 29, 2026
1f2c73a
Add field-line tracer tests
abhiyan123 Oct 29, 2026
1b11d3c
Add team barriers to compact bicubic spline setup
abhiyan123 Oct 29, 2026
dc0436a
Merge branch 'fix-bicubic-spline-team-barrier' into field-line-tracer
abhiyan123 Oct 29, 2026
9c26bd8
Add more field-line tracer tests
abhiyan123 Oct 29, 2026
1ded4a9
Align spline barrier placement with #397
abhiyan123 Oct 29, 2026
8bded2b
Merge branch 'fix-bicubic-spline-team-barrier' into field-line-tracer
abhiyan123 Oct 29, 2026
77b9a6e
Use fseek instead of rewind in field-line tracer test
abhiyan123 Oct 29, 2026
f45610c
Add non-square off-axis equilibrium test for field-line tracer
abhiyan123 Oct 29, 2026
def55dc
Allocate RK4 work views once per field-line trace
abhiyan123 Oct 29, 2026
ec75341
test poloidal return and reference trace
abhiyan123 Oct 30, 2026
3834e90
explain tracer tests
abhiyan123 Oct 30, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/pcms/field/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,8 @@ if (PCMS_ENABLE_OMEGA_H)
layout/omega_h_lagrange.cpp
layout/uniform_grid.cpp
evaluator/mls_interpolation.cpp
field_aligned/magnetic_field.cpp
field_aligned/field_line_tracer.cpp
)
list(APPEND PCMS_FIELD_HEADERS
function_space/lagrange.h
Expand All @@ -66,6 +68,8 @@ if (PCMS_ENABLE_OMEGA_H)
evaluator/pcms_interpolator_logger.hpp
evaluator/mls_point_cloud.h
evaluator/point_cloud.h
field_aligned/magnetic_field.hpp
field_aligned/field_line_tracer.hpp
)
endif ()

Expand Down
6 changes: 6 additions & 0 deletions src/pcms/field/evaluator/spline_interpolator.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1892,6 +1892,7 @@ void solve_bicubic_spline_compact(Rank1View<T, MemorySpace> x, LO nx,
Kokkos::parallel_for(Kokkos::TeamThreadRange(team, nx),
[=](LO ix) { fwk_x_view(0, ix) = f(0, ix, iy); });

team.team_barrier();
if (team.team_rank() == 0) {
if (ibcxmin == 1 || ibcxmin == 2)
zbcmin = bcxmin[iy];
Expand All @@ -1901,6 +1902,7 @@ void solve_bicubic_spline_compact(Rank1View<T, MemorySpace> x, LO nx,
solve_cubic_spline_compact(x, nx, fwk_x_view, fwk4_x_view, ibcxmin,
zbcmin, ibcxmax, zbcmax, wk_x_view);
}
team.team_barrier();

Kokkos::parallel_for(Kokkos::TeamThreadRange(team, nx),
[=](LO ix) { f(1, ix, iy) = fwk_x_view(1, ix); });
Expand All @@ -1922,6 +1924,7 @@ void solve_bicubic_spline_compact(Rank1View<T, MemorySpace> x, LO nx,
Kokkos::parallel_for(Kokkos::TeamThreadRange(team, ny),
[=](LO iy) { fwk_y_view(0, iy) = f(0, ix, iy); });

team.team_barrier();
if (team.team_rank() == 0) {
if (iflg2 == 1) {
if (ibcymin == 1 || ibcymin == 2)
Expand All @@ -1933,6 +1936,7 @@ void solve_bicubic_spline_compact(Rank1View<T, MemorySpace> x, LO nx,
solve_cubic_spline_compact(y, ny, fwk_y_view, fwk4_y_view, ibcmin, 0.0,
ibcmax, 0.0, wk_y_view);
}
team.team_barrier();
Kokkos::parallel_for(Kokkos::TeamThreadRange(team, ny),
[=](LO iy) { f(2, ix, iy) = fwk_y_view(1, iy); });
});
Expand All @@ -1953,6 +1957,7 @@ void solve_bicubic_spline_compact(Rank1View<T, MemorySpace> x, LO nx,
Kokkos::parallel_for(Kokkos::TeamThreadRange(team, ny),
[=](LO iy) { fwk_y_view(0, iy) = f(1, ix, iy); });

team.team_barrier();
if (team.team_rank() == 0) {
if (iflg2 == 1) {
if (ibcymin == 1 || ibcymin == 2)
Expand All @@ -1964,6 +1969,7 @@ void solve_bicubic_spline_compact(Rank1View<T, MemorySpace> x, LO nx,
solve_cubic_spline_compact(y, ny, fwk_y_view, fwk4_y_view, ibcmin, 0.0,
ibcmax, 0.0, wk_y_view);
}
team.team_barrier();
Kokkos::parallel_for(Kokkos::TeamThreadRange(team, ny),
[=](LO iy) { f(3, ix, iy) = fwk_y_view(1, iy); });
});
Expand Down
Loading
Loading