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

Test GTIR-DaCe backend #638

Draft
wants to merge 133 commits into
base: main
Choose a base branch
from
Draft
Changes from 1 commit
Commits
Show all changes
133 commits
Select commit Hold shift + click to select a range
f346ef1
Use new itir.Program everywhere
tehrengruber Nov 14, 2024
a80a1e0
Merge remote-tracking branch 'origin/main' into update_to_gtir
tehrengruber Dec 6, 2024
61e97e4
Use gt4py main again
tehrengruber Dec 6, 2024
416d7e7
fix connectivities
havogt Dec 9, 2024
38a162c
switch gt4py branch
havogt Dec 9, 2024
21c495d
fix more connectivities
edopao Dec 20, 2024
0e8fcba
fix more connectivities (1)
edopao Dec 20, 2024
d784012
Merge remote-tracking branch 'origin/main' into update_to_gtir
edopao Dec 20, 2024
7917c51
update versions (temporarily)
havogt Jan 8, 2025
e7e5a03
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Jan 10, 2025
64b12d3
switch gt4py branch to dace-gtir-scan
edopao Jan 10, 2025
150d48c
update uv lock
edopao Jan 10, 2025
c26c235
update dace CI-config
edopao Jan 10, 2025
b627c71
pytest marker for tests that require concat_where
edopao Jan 10, 2025
b80e5b7
update lock file
edopao Jan 10, 2025
042f7dd
disable orchestration tests
edopao Jan 10, 2025
78d81ce
disable orchestration tests (1)
edopao Jan 10, 2025
84f1ec7
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Jan 10, 2025
363cc2a
update uv lock
edopao Jan 13, 2025
6481b65
enable orchestration test cases in diffusion module
edopao Jan 13, 2025
e78cf66
fix parameterization of dace orchestration
edopao Jan 13, 2025
d15c2f6
update uv lock
edopao Jan 13, 2025
f0eca78
Fix some errors in orchestration decorator
edopao Jan 13, 2025
8307f04
update orchestrator decorator for gtir
edopao Jan 13, 2025
242850e
update uv lock
edopao Jan 13, 2025
3624cd2
DaCe Orchestration: WIP
kotsaloscv Jan 14, 2025
6c57fb2
workaround for compile_time_connectivities
edopao Jan 14, 2025
e78f239
add gt4py_cache to gitignore
edopao Jan 14, 2025
95eef9c
update uv lock
edopao Jan 14, 2025
195cba6
update dace check for diffusion orchestration
edopao Jan 15, 2025
eb50015
update uv lock
edopao Jan 15, 2025
d3933a4
Update pyproject.toml
havogt Jan 16, 2025
ad1574d
update uv lock
edopao Jan 16, 2025
6e2d866
update test marker
edopao Jan 21, 2025
35a52bd
re-enable fused_velocity_advection_stencil_8_to_13 test
edopao Jan 22, 2025
922f191
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Jan 22, 2025
0c0cd70
switch to gt4py main
edopao Jan 22, 2025
9a2c63e
fix gt4py import
edopao Jan 22, 2025
b404e09
switch gt4py to gtir-dace
edopao Jan 22, 2025
d736f6a
update uv lock
edopao Jan 22, 2025
ffc94e3
update uv lock
edopao Jan 23, 2025
dbcdc5b
fix for dace orchestration
edopao Jan 23, 2025
7a6370c
update uv lock
edopao Jan 23, 2025
bc6a335
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Jan 24, 2025
533ed87
update uv lock
edopao Jan 24, 2025
73b5bbb
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Jan 24, 2025
a56bd26
update uv lock
edopao Jan 24, 2025
3cd9704
enable dace_gpu on test_model_datatests
edopao Jan 24, 2025
6c1e72d
update uv lock
edopao Jan 24, 2025
09804e9
update uv lock
edopao Jan 24, 2025
804e110
Merge remote-tracking branch 'upstream/main' into update_to_gtir
havogt Jan 27, 2025
b02a659
lock
havogt Jan 27, 2025
7a532be
point to new branch, fix table access
havogt Jan 27, 2025
8a5986d
update uv lock
edopao Jan 28, 2025
4d2dda2
update uv lock
edopao Jan 28, 2025
f1c2722
edit DummyNestedSDFG in dace orchestrator
edopao Jan 29, 2025
3a8f510
update uv lock
edopao Jan 29, 2025
e5d3364
edit orchestration decorator
edopao Jan 29, 2025
0b4e5a4
edit orchestration decorator
edopao Jan 29, 2025
41fdbd0
Enhance nox and CI configuration
egparedes Jan 30, 2025
8cd12b1
Support empty collection of tests
egparedes Jan 30, 2025
1b8789d
Merge branch 'main' into enhance-nox-config
egparedes Jan 30, 2025
d07bcf2
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Jan 31, 2025
6d37933
update uv lock
edopao Jan 31, 2025
b81edfc
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Jan 31, 2025
e039f9c
Re-enable diffusion tests when orchestration is False
edopao Jan 31, 2025
11c94e1
Merge remote-tracking branch 'origin/fix_diffusion_tests' into update…
edopao Jan 31, 2025
8ed37fb
fix pre-commit
edopao Jan 31, 2025
65b7335
disable orchestration tests on GPU
edopao Jan 31, 2025
9cfa967
disable embedded backend on diffusion tests
edopao Jan 31, 2025
544bad0
Merge remote-tracking branch 'origin/fix_diffusion_tests' into update…
edopao Jan 31, 2025
b743feb
skipping fused_velocity_advection_stencil_1_to_7 because of concat_where
edopao Feb 4, 2025
2eddb27
switch to dace main from git repo
edopao Feb 4, 2025
8536faf
change xfail to skip to avoid segfault
edopao Feb 4, 2025
cf1f0cb
update uv lock
edopao Feb 4, 2025
f753277
switch to dace main
edopao Feb 4, 2025
039f338
fix for dace orchestration
edopao Feb 4, 2025
7ef629f
update uv lock
edopao Feb 4, 2025
2bc3a08
Merge origin/main
tehrengruber Feb 4, 2025
81eeebe
Merge remote-tracking branch 'refs/remotes/origin/update_to_gtir' int…
tehrengruber Feb 4, 2025
afd6c94
Small fixes
tehrengruber Feb 5, 2025
1321686
update uv lock
edopao Feb 5, 2025
5a0606a
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Feb 5, 2025
478ab6f
fix for test marker checker
edopao Feb 5, 2025
e64d3a0
update uv lock
edopao Feb 5, 2025
1a904fa
re-enable combined stencils
edopao Feb 6, 2025
2dbf117
update uv lock
edopao Feb 6, 2025
6b59ade
switch to gt4py main
edopao Feb 7, 2025
5b35053
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Feb 11, 2025
f1e4b1d
update uv lock
edopao Feb 11, 2025
da77ac5
update uv lock
edopao Feb 13, 2025
9f64c14
echo test command
edopao Feb 13, 2025
376274a
Merge remote-tracking branch 'origin/main' into update_to_gtir
tehrengruber Feb 14, 2025
b36330b
Small fixes
tehrengruber Feb 14, 2025
424c3ff
Upgrade ubuntu image used in CI (required for gt4py)
tehrengruber Feb 14, 2025
87dc687
Use staging branch
tehrengruber Feb 14, 2025
40c6523
Fix dependency for ubuntu 22.04
tehrengruber Feb 14, 2025
0855efc
Small fies
tehrengruber Feb 14, 2025
de4d301
Small fixes
tehrengruber Feb 14, 2025
230f7f4
Small fixes
tehrengruber Feb 14, 2025
7f0900b
Small fixes
tehrengruber Feb 14, 2025
3c38f73
Merge remote-tracking branch 'origin/update_to_gtir' into update_to_g…
edopao Feb 14, 2025
b9bd0a4
update uv lock
edopao Feb 14, 2025
f8ee082
nox debug print
edopao Feb 14, 2025
f6c89da
Merge remote-tracking branch 'origin/main' into enhance-nox-config
edopao Feb 14, 2025
a4e8a88
remove nox debug print
edopao Feb 14, 2025
5d60bc1
Merge branch 'main' into enhance-nox-config
edopao Feb 14, 2025
9983abc
Merge remote-tracking branch 'origin/enhance-nox-config' into update_…
edopao Feb 14, 2025
fd95ebc
switch gt4py to main
edopao Feb 14, 2025
284cdda
remove extra changes
edopao Feb 14, 2025
7370478
fix gitlab config
edopao Feb 14, 2025
fdb0ea8
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Feb 14, 2025
e5e4718
fix nox component
edopao Feb 14, 2025
22ec35a
change component selection in nox config
edopao Feb 15, 2025
602ba32
skip orchestration tests on gpu
edopao Feb 15, 2025
ef0ff4f
add nox debug print
edopao Feb 17, 2025
09e95cb
remove extra change
edopao Feb 17, 2025
72878a2
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Feb 17, 2025
7dbaf59
read env variable from os environment, not from nox
edopao Feb 17, 2025
aceb53d
Merge remote-tracking branch 'origin/fix-ci' into update_to_gtir_dace
edopao Feb 17, 2025
bc2cb81
update uv lock
edopao Feb 17, 2025
827edaa
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Feb 17, 2025
bcbe854
Merge branch 'main' into fix_diffusion_tests
edopao Feb 17, 2025
f500f93
import fix from diffusion granule
edopao Feb 17, 2025
7d384c7
Revert "import fix from diffusion granule"
edopao Feb 17, 2025
eecf2a3
update uv lock
edopao Feb 17, 2025
fb4ac89
Merge remote-tracking branch 'origin/main' into fix_diffusion_tests
edopao Feb 18, 2025
353da63
Merge remote-tracking branch 'origin/main' into update_to_gtir_dace
edopao Feb 18, 2025
990f4e3
adjust tolerance in diffusion test
edopao Feb 18, 2025
32b1ac2
Merge remote-tracking branch 'origin/fix_diffusion_tests' into update…
edopao Feb 18, 2025
bacd119
update uv lock
edopao Feb 18, 2025
93a5881
review comments
edopao Feb 19, 2025
503dfcf
Merge remote-tracking branch 'origin/fix_diffusion_tests' into update…
edopao Feb 19, 2025
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
Prev Previous commit
Next Next commit
fix more connectivities (1)
edopao committed Dec 20, 2024
commit 0e8fcba34777f4cc6f5c8328b33993690e768d56
Original file line number Diff line number Diff line change
@@ -26,7 +26,7 @@ def interpolate_to_cell_center_numpy(
grid, interpolant: np.array, e_bln_c_s: np.array, **kwargs
) -> np.array:
e_bln_c_s = np.expand_dims(e_bln_c_s, axis=-1)
c2ce = grid.get_offset_provider("C2CE").table
c2ce = grid.get_offset_provider("C2CE").ndarray

interpolation = np.sum(
interpolant[grid.connectivities[dims.C2EDim]] * e_bln_c_s[c2ce],