Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
62 commits
Select commit Hold shift + click to select a range
c247d41
Add prototype support for for loops over data dims.
stubbiali Oct 3, 2024
2529e02
Merge remote-tracking branch 'upstream/main' into data-dim-loops
stubbiali Oct 3, 2024
cc2def3
Add support for lists and tuples.
stubbiali Oct 4, 2024
e79917a
Cosmetics.
stubbiali Oct 24, 2024
0d321ec
Allow field data indices.
stubbiali Oct 24, 2024
0856861
Call CallInliner before DataDimLoopUnroller.
stubbiali Oct 25, 2024
4847fb4
Casting to INT. Add `v_in_int = int(v_in_float)` op as a base unitary…
FlorianDeconinck Aug 19, 2024
8f5fe95
Lint
Oct 10, 2024
a70a46c
Native function: f32, f64, round
FlorianDeconinck Oct 24, 2024
1d7cbad
Enable for-loops in functions.
stubbiali Oct 25, 2024
511222a
Enable functions with no return statements.
stubbiali Oct 25, 2024
45363a3
Allow assigning call arguments inside functions.
stubbiali Oct 25, 2024
7015268
Add inlining of constant function arguments.
stubbiali Nov 21, 2024
d151a78
Inline constant function arguments recursively.
stubbiali Nov 22, 2024
10c1bfb
Properly support for-loops around functions.
stubbiali Nov 22, 2024
90743af
Merge remote-tracking branch 'upstream/main' into ecrad
stubbiali Nov 22, 2024
c811ba5
Improve support for global tables in numpy generated code.
stubbiali Dec 19, 2024
5bd2ec2
Merge remote-tracking branch 'upstream/main' into ecrad
stubbiali Dec 19, 2024
878fd35
Fully inline constant arguments.
stubbiali Jan 27, 2025
631ed51
Merge remote-tracking branch 'upstream/main' into ecrad
stubbiali Jan 30, 2025
4f25fa3
Prototype implementation of reductions.
stubbiali Feb 4, 2025
a75a0de
Fix numpy codegen.
stubbiali Mar 5, 2025
01e7555
Fix "TypeError: : cannot pickle 'PyCapsule' object".
stubbiali May 20, 2025
1230920
Add custom AST nodes.
stubbiali May 21, 2025
a5398fb
Add For and ForIndex defir nodes.
stubbiali May 21, 2025
1e65bd5
Add custom For and ForIndex gtir nodes.
stubbiali May 21, 2025
d13a8ab
Refactor common ForIndex.
stubbiali May 21, 2025
c8dd831
Add For and ForIndex oir nodes.
stubbiali May 21, 2025
6ed7ddd
Add For and ForIndex gtcpp nodes.
stubbiali May 21, 2025
8570ce1
Support For and ForIndex in codegen.
stubbiali May 21, 2025
de465c6
Make For and ForIndex deep-copyable.
stubbiali May 22, 2025
cf0d34b
index -> index_name
stubbiali May 22, 2025
9af5a5c
Add support for for-loops in dace.
stubbiali May 23, 2025
011e3d4
Fix single precision.
stubbiali May 23, 2025
57d7f5f
Fix cuda extra compile args
stubbiali Apr 7, 2026
36c24fa
Add support for for-loops in numpy backend
stubbiali Apr 13, 2026
a1de48f
Merge remote-tracking branch 'upstream/main' into loops-over-data-dim…
stubbiali Jun 5, 2026
cfa85f4
avoid int overflow for arrays with many gridpoints
gvollenweider Jun 6, 2026
d52cb9a
Merge remote-tracking branch 'upstream/main' into loops-over-data-dims
stubbiali Jun 30, 2026
3520e9e
Merge remote-tracking branch 'upstream/main' into loops-over-data-dims
stubbiali Jul 3, 2026
ef219b6
Merge remote-tracking branch 'upstream/main' into loops-over-data-dims
stubbiali Jul 31, 2026
a61087d
Merge remote-tracking branch 'upstream/main' into loops-over-data-dims
stubbiali Sep 10, 2026
41300dc
Pass extra cxx compile args to cuda/rocm compiler
stubbiali Sep 10, 2026
e5805fd
Revert "feat[cartesian]: DaCe optimal for/map schedule (#2628)"
stubbiali Sep 21, 2026
b04e352
Revert "refactor[cartesian]: update treeir representation of the vert…
stubbiali Sep 21, 2026
62b0896
Add gt4py and dace version to stencil fingerprint
stubbiali Sep 21, 2026
f6d0c10
Merge remote-tracking branch 'upstream/main' into loops-over-data-dims
stubbiali Sep 21, 2026
06160a4
Fix vertical loop index symbol
stubbiali Sep 21, 2026
25f7d66
Try moving vertical loop inside the kernel
stubbiali Sep 21, 2026
29d4201
Revert "Try moving vertical loop inside the kernel"
stubbiali Sep 23, 2026
bd91ef7
Reapply "refactor[cartesian]: update treeir representation of the ver…
stubbiali Sep 23, 2026
00e9999
Reapply "feat[cartesian]: DaCe optimal for/map schedule (#2628)"
stubbiali Sep 23, 2026
5ad3321
Parametrize parallel vertical loop schedule
stubbiali Sep 23, 2026
e031aa7
Apply map collapse transformation to sdgf
stubbiali Sep 23, 2026
b41448f
Fix typo
stubbiali Sep 23, 2026
92923b1
Remove debug code
stubbiali Sep 23, 2026
c02a484
Merge remote-tracking branch 'origin/fix-dace-perf' into loops-over-d…
stubbiali Sep 23, 2026
9d6749f
Merge remote-tracking branch 'upstream/main' into fix-dace-perf
stubbiali Sep 30, 2026
b9926f4
Do not validate map-collapse pass
stubbiali Sep 30, 2026
19686d1
Fix unit tests
stubbiali Sep 30, 2026
82796c8
Merge remote-tracking branch 'origin/fix-dace-perf' into loops-over-d…
stubbiali Sep 30, 2026
dca1458
Merge remote-tracking branch 'upstream/main' into loops-over-data-dims
stubbiali Oct 2, 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
5 changes: 5 additions & 0 deletions src/gt4py/cartesian/backend/pyext_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,11 @@ def get_gt_pyext_build_opts(
"-std=c++20",
f"-ftemplate-depth={gt_config.build_settings['cpp_template_depth']}",
*extra_compile_args_from_config["cuda"],
*(
arg
for cxx_compile_arg in extra_compile_args["cxx"]
for arg in f"--compiler-options {cxx_compile_arg}".split(" ")
),
]
if is_rocm_gpu:
extra_compile_args["cuda"] += [
Expand Down
15 changes: 15 additions & 0 deletions src/gt4py/cartesian/frontend/defir_to_gtir.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,8 @@
Expr,
FieldDecl,
FieldRef,
For,
ForIndex,
HorizontalIf,
If,
IterationOrder,
Expand Down Expand Up @@ -582,6 +584,19 @@ def visit_While(self, node: While) -> gtir.While:
loc=location_to_source_location(node.loc),
)

def visit_ForIndex(self, node: ForIndex) -> gtir.ForIndex:
return gtir.ForIndex(name=node.name, dtype=common.DataType(node.data_type.value))

def visit_For(self, node: For) -> gtir.For:
return gtir.For(
index_name=node.index_name,
iter_start=node.iter_start,
iter_stop=node.iter_stop,
iter_step=node.iter_step,
body=self.visit(node.body),
loc=location_to_source_location(node.loc),
)

def visit_VarRef(self, node: VarRef, **kwargs) -> gtir.ScalarAccess:
return gtir.ScalarAccess(name=node.name, loc=location_to_source_location(node.loc))

Expand Down
Loading
Loading