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

feat[next][dace]: Parallel Map Fusion #1620

Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
258 commits
Select commit Hold shift + click to select a range
daf7827
Remove hard-coded field shape
edopao Apr 23, 2024
9672b3b
Remove hard-coded target domain
edopao Apr 23, 2024
b6326b8
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao Apr 23, 2024
26f3790
Refactoring
edopao Apr 24, 2024
1efffa7
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao Apr 24, 2024
f99fa84
Fix formatting
edopao Apr 24, 2024
d6e1088
More refactoring
edopao Apr 24, 2024
9854497
Minor edit
edopao Apr 24, 2024
37d83d7
Fix formatting
edopao Apr 24, 2024
29986ef
Use callable to build taskgraph
edopao Apr 29, 2024
390f3b4
Add draft of select operator
edopao Apr 29, 2024
de27419
Remove node mapping
edopao Apr 30, 2024
cd900f5
Remove node mapping (fix + test case)
edopao Apr 30, 2024
326cbb5
Add test case for inlined mathematic builtins
edopao Apr 30, 2024
a10b614
Go full functional (remove SDFGState member var)
edopao Apr 30, 2024
aef4265
Minor edit
edopao Apr 30, 2024
9e67dfe
Minor edit (1)
edopao Apr 30, 2024
4b4109e
Fix state handling
edopao Apr 30, 2024
495fd0a
Edit comments based on review
edopao Apr 30, 2024
0085194
Add test case for nested select
edopao Apr 30, 2024
41e2a44
Separate builtin translation from driver logic
edopao May 1, 2024
7148c5f
Improve code comments
edopao May 2, 2024
452399d
Avoid inheritance: pass dataflow builder as arg to builtin translator
edopao May 3, 2024
e404226
Codestyle review changes
edopao May 3, 2024
bb0dfac
Remove circular dependency for builtin translators
edopao May 6, 2024
412cd5d
Fix formatting
edopao May 6, 2024
651de5c
Minor edit
edopao May 6, 2024
dcf3eab
Add support to translate each builtin call to a tasklet node
edopao May 6, 2024
7e6909e
Resolve dace warnings
edopao May 7, 2024
2b07cc5
Remove bultin translator for domain expressions
edopao May 7, 2024
2370fa6
Remove bultin translator for domain expressions (1)
edopao May 7, 2024
8e801df
Refactor
edopao May 7, 2024
812a6e5
Minor edit
edopao May 7, 2024
1d0b50b
Extract ITIR visitor to separate class
edopao May 7, 2024
97a1d22
Code refactoring
edopao May 7, 2024
a30cc7d
Fix formatting
edopao May 7, 2024
f595b01
Add IteratorExpr type
edopao May 10, 2024
a6bcb6c
Indirection shift implemented as tasklet node
edopao May 10, 2024
738da27
Add ConnectivityExpr type
edopao May 13, 2024
e5494d8
Remove ConnectivityExpr type, use ValueExpr instead
edopao May 13, 2024
e9455e3
Changes in preparation for shift builtin
edopao May 13, 2024
cbf55de
Refactoring
edopao May 13, 2024
9d5b1ed
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 13, 2024
801704b
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao May 13, 2024
c45c417
Add support for programs without computation (pure memlets)
edopao May 13, 2024
3f26d91
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 13, 2024
f173244
Merge remote-tracking branch 'origin/main' into dace-fieldview-shifts
edopao May 13, 2024
d67518a
Fix test
edopao May 13, 2024
783542f
Fix for chain of shift expressions shift(V2E(E2V(i_edge, x), y))(edges)
edopao May 14, 2024
1fa9de4
Support for multi-dimensional shift
edopao May 14, 2024
96338c2
Fix typo
edopao May 14, 2024
57e369f
Add support for cartesian shift with dynamic offset
edopao May 15, 2024
ec4714c
Add support for unstructured shift with dynamic offset
edopao May 15, 2024
46cb6c6
Code refactoring in test file
edopao May 15, 2024
c20a94d
Typo
edopao May 15, 2024
d1f7432
Code cleanup
edopao May 15, 2024
c4385c1
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao May 16, 2024
ed16fd4
Import updates from branch dace-fieldview-shifts
edopao May 16, 2024
9f7176f
Review comments
edopao May 16, 2024
4f40f42
Merge branch 'dace-fieldview' into dace-fieldview-shifts
edopao May 16, 2024
932db7c
Avoid tasklet-to-tasklet edge connections
edopao May 16, 2024
46febb0
Avoid tasklet-to-tasklet edge connections
edopao May 16, 2024
949bad7
Add support for in-out field parameters
edopao May 16, 2024
8890f95
Refactoring: import modules, not symbols
edopao May 17, 2024
87b71a6
Minor edit
edopao May 17, 2024
665a609
Remove internal package for builtin translators
edopao May 17, 2024
82fdf64
Add wrapper function to build SDFG
edopao May 17, 2024
e4718b0
Merge pull request #4 from edopao/dace-fieldview-refactor_imports
edopao May 17, 2024
47fcabe
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 17, 2024
51aaf0f
Add fieldview flavor of all test cases
edopao May 17, 2024
6ccecf1
Code changes imported from branch dace-fieldview-shifts
edopao May 17, 2024
e66b960
Code comments updated
edopao May 17, 2024
7f89a16
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 17, 2024
4c190bd
Remove support for inlined chained shift
edopao May 21, 2024
6052de2
Add support for neighbors builtin
edopao May 21, 2024
7300864
Add support for reduce builtin
edopao May 22, 2024
55adbd5
Refactoring
edopao May 23, 2024
ad21dc4
Add support for both inlined and fieldview neighbor reduction
edopao May 23, 2024
bb9123b
Minor edit
edopao May 23, 2024
0025d77
Code refactoring
edopao May 23, 2024
9926d7d
Add support for skip values ONLY for inlined GTIR
edopao May 23, 2024
172f19e
Masked array implementation based on connectivity table
edopao May 27, 2024
b1f4a47
Merge 2 different implementations of reduce
edopao May 27, 2024
63e6e92
Add support for reduce lambda function
edopao May 28, 2024
107e295
Add support for neighbors masked array returned by select statements
edopao May 29, 2024
3c71efa
Import changes from neighbors branch
edopao May 29, 2024
e369cac
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 29, 2024
d0bd277
Import changes from neighbors branch
edopao May 29, 2024
afb5ed1
Merge remote-tracking branch 'origin/dace-fieldview-shifts' into dace…
edopao May 29, 2024
2f75cfb
Add debuginfo for ir.Program and ir.Stmt nodes
edopao May 29, 2024
695db7c
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 29, 2024
074f0b2
Merge remote-tracking branch 'origin/dace-fieldview-shifts' into dace…
edopao May 29, 2024
085f307
Fix error in debuginfo
edopao May 29, 2024
f19960b
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao May 29, 2024
841040e
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 29, 2024
b3df358
Merge remote-tracking branch 'origin/dace-fieldview-shifts' into dace…
edopao May 29, 2024
dc1434c
Fix error in debuginfo (1)
edopao May 29, 2024
eacde66
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao May 29, 2024
138a33c
Merge remote-tracking branch 'origin/dace-fieldview-shifts' into dace…
edopao May 29, 2024
3769fb5
Remove nested SDFG for neighbors builtin
edopao Jun 14, 2024
b1b5887
Remove masked array for skip values, rely on identity value
edopao Jun 26, 2024
a5b0f41
import changes from neighbors branch
edopao Jun 28, 2024
f7ac3d8
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao Jun 28, 2024
01ff262
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao Jun 28, 2024
c61e796
import changes from neighbors branch
edopao Jun 28, 2024
5a457b2
Merge remote-tracking branch 'origin/dace-fieldview-shifts' into dace…
edopao Jun 28, 2024
f4d9d89
Let's see what auto opt can do.
philip-paul-mueller Jul 3, 2024
9318011
Import changes from branch dace-fieldview-neighbors
edopao Jul 4, 2024
11efdeb
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao Jul 4, 2024
25b9048
Support field with start offset
edopao Jul 4, 2024
2dc6f97
Merge branch 'dace-fieldview' into dace-fieldview-shifts
edopao Jul 4, 2024
f6e5b7c
Add test coverage for temporary with start offset (cartesian shift)
edopao Jul 4, 2024
d7312fa
Support field with start offset
edopao Jul 4, 2024
628c18b
Merge branch 'dace-fieldview' into dace-fieldview-shifts
edopao Jul 4, 2024
c4f2738
Test IR updated for literal operand
edopao Jul 4, 2024
0fd0b65
Add test coverage to previous commit
edopao Jul 4, 2024
38d2720
Refactor PrimitiveTranslator interface
edopao Jul 4, 2024
d3541c1
Made a small modfication to some code.
philip-paul-mueller Jul 5, 2024
e855ef9
Fix formatting
edopao Jul 5, 2024
5726509
Started with a first nabla stuff.
philip-paul-mueller Jul 5, 2024
e44f3a2
It seems that local storage does not work well with this transformer.
philip-paul-mueller Jul 5, 2024
4cff071
Fix for domain horzontal/vertical dims
edopao Jul 5, 2024
f642e85
Fix for type inference on single value expression
edopao Jul 5, 2024
f216a36
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao Jul 5, 2024
a2af8cd
Merge remote-tracking branch 'edoardo/dace-fieldview' into dace-field…
philip-paul-mueller Jul 5, 2024
74bd468
Updated it now seems to work.
philip-paul-mueller Jul 5, 2024
667eb7e
Updated the nabla4 calculations.
philip-paul-mueller Jul 5, 2024
58b8e58
Now all the calculations are done.
philip-paul-mueller Jul 5, 2024
e898b31
Formated a bit.
philip-paul-mueller Jul 5, 2024
eae968f
Refactored the code.
philip-paul-mueller Jul 5, 2024
defb55d
Import changes from dace-fieldview-neighbors
edopao Jul 5, 2024
fc9661c
Import changes from dace-fieldview-shifts
edopao Jul 5, 2024
e424d4e
Minor edit
edopao Jul 5, 2024
7ef1d56
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao Jul 5, 2024
563ee1a
WIP: Working on accessing.
philip-paul-mueller Jul 7, 2024
0dc376e
Merge remote-tracking branch 'edoardo/dace-fieldview' into dace-field…
philip-paul-mueller Jul 8, 2024
9df80ad
Merge remote-tracking branch 'edoardo/dace-fieldview-shifts' into dac…
philip-paul-mueller Jul 8, 2024
f32fd38
Now the shift works, at least the shift in the particular dimension.
philip-paul-mueller Jul 8, 2024
538abff
Prepare to go to real input.
philip-paul-mueller Jul 8, 2024
a07fe81
nabla4 works now with the custom icon stuff.
philip-paul-mueller Jul 8, 2024
fec054a
First step in shifting.
philip-paul-mueller Jul 8, 2024
ea7bf64
Now we have one shifting.
philip-paul-mueller Jul 8, 2024
b291152
The helper function works.
philip-paul-mueller Jul 8, 2024
008209d
It now works with the normal shiftuing stuff.
philip-paul-mueller Jul 8, 2024
b832aca
Now the full nabla4 should be ported.
philip-paul-mueller Jul 8, 2024
94ab9d7
Restructured the code and removed the inline version.
philip-paul-mueller Jul 8, 2024
04cde84
Made some small update.
philip-paul-mueller Jul 8, 2024
3dd0860
This is the base of all fusion operations.
philip-paul-mueller Jul 9, 2024
3178b71
Reworked some parts.
philip-paul-mueller Jul 10, 2024
66c5fcd
Address review comments
edopao Jul 10, 2024
d5abad4
Merge remote-tracking branch 'origin/main' into dace-fieldview
edopao Jul 10, 2024
1df1bc3
Apply convention for map variables
edopao Jul 10, 2024
2032b60
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao Jul 10, 2024
6394243
Updated and fixed a big in the `is_interstate_transient()` function.
philip-paul-mueller Jul 10, 2024
fcd8ee3
Small corrections and format improvements.
philip-paul-mueller Jul 11, 2024
a25a6a4
Fixed some missing include.
philip-paul-mueller Jul 11, 2024
a57e108
Added a first and mostly untested version of the serial fusion transf…
philip-paul-mueller Jul 11, 2024
62ad165
Started debugin, very strange bug.
philip-paul-mueller Jul 11, 2024
7f72794
Import changes from dace-fieldview-neighbors
edopao Jul 11, 2024
abf3918
Import changes from dace-fieldview-shifts
edopao Jul 11, 2024
a6d31fb
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao Jul 11, 2024
4a2ccaa
More debugger friendly.
philip-paul-mueller Jul 11, 2024
d353d0e
It should now work and I figured out why it was not working before.
philip-paul-mueller Jul 11, 2024
073065d
A fix.
philip-paul-mueller Jul 11, 2024
19be2c4
Added some "test" for the merger.
philip-paul-mueller Jul 11, 2024
5237b13
Now the nabla4 optimizes with my fusion operation.
philip-paul-mueller Jul 11, 2024
489bb4a
Added some more test.
philip-paul-mueller Jul 12, 2024
2f6274e
Merge remote-tracking branch 'edoardo/dace-fieldview' into dace-field…
philip-paul-mueller Jul 12, 2024
4d1a3cc
Fixed some small problem in detecting recursive dataflow.
philip-paul-mueller Jul 12, 2024
2da7453
Made some imporvements to the test.
philip-paul-mueller Jul 12, 2024
ba97fd2
Made some comments better.
philip-paul-mueller Jul 12, 2024
9301dbe
Import changes from branch dace-fieldview-neighbors
edopao Jul 12, 2024
7f60cfe
Import changes from branch dace-fieldview-shifts
edopao Jul 12, 2024
699a88b
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao Jul 12, 2024
b3131db
Avoid direct import of symbols from module
edopao Jul 12, 2024
130c877
Address review comments
edopao Jul 12, 2024
7fbd7e1
Merge remote-tracking branch 'origin/dace-fieldview' into dace-fieldv…
edopao Jul 12, 2024
fb2ba90
Started with an untested map promoted.
philip-paul-mueller Jul 12, 2024
bf06cb4
Merge remote-tracking branch 'edoardo/dace-fieldview' into dace-field…
philip-paul-mueller Jul 12, 2024
52c1d01
Updated the tests, but it still does not work.
philip-paul-mueller Jul 12, 2024
849900f
The newest version about shift from Edoardo.
philip-paul-mueller Jul 12, 2024
42f4aba
Now the shift test works too.
philip-paul-mueller Jul 12, 2024
84b2ba7
Added some more checking functionality to the base promoter.
philip-paul-mueller Jul 12, 2024
24bde91
Added a concrete promoter.
philip-paul-mueller Jul 12, 2024
fd81e75
Added a custom (okay currently not really custom) simplification pass.
philip-paul-mueller Jul 12, 2024
284b6a8
Updated the auto fusion stuff.
philip-paul-mueller Jul 12, 2024
a6b191c
Merge remote-tracking branch 'origin/main' into dace-fieldview-shifts
edopao Jul 12, 2024
033db6b
Removed all my non gt4py parts and moved it to a separate repo. The t…
philip-paul-mueller Jul 15, 2024
32712ea
Merge remote-tracking branch 'gt4py/main' into dace-fieldview-transfo…
philip-paul-mueller Jul 15, 2024
ebb76de
Merge remote-tracking branch 'edoardo/dace-fieldview-shifts' into dac…
philip-paul-mueller Jul 15, 2024
0fddb8d
Added a transformation to bring the map iteration indexes in the corr…
philip-paul-mueller Jul 15, 2024
c4f64a4
Updated the auto optimizer.
philip-paul-mueller Jul 15, 2024
6143b95
Made the `gt_simplify()` function aviable.
philip-paul-mueller Jul 17, 2024
00aa64c
Fixed a porting bug.
philip-paul-mueller Jul 17, 2024
b67f0c0
Fixed an edge case in the computation of the output partition if teh …
philip-paul-mueller Jul 17, 2024
b447c2a
Added a map promoter that is able to promote trivial maps that are ge…
philip-paul-mueller Jul 17, 2024
090f08d
Added a function to turn an SDFG into one that runs on GPU.
philip-paul-mueller Jul 17, 2024
04dd63a
Updated the auto optimizer to handle GPU cases.
philip-paul-mueller Jul 17, 2024
bb34f44
Reorganized the GPU stuff.
philip-paul-mueller Jul 18, 2024
88f5245
There is now a k blocking transformation.
philip-paul-mueller Jul 19, 2024
73a01c1
Made some fixes to the k blocking stuff.
philip-paul-mueller Jul 19, 2024
dd1242c
Small fix.
Jul 19, 2024
f3798f3
Fixed an error.
philip-paul-mueller Jul 19, 2024
863bd5f
Now auto optimization also does blocking.
philip-paul-mueller Jul 19, 2024
ea0da2b
Made some fixes, but it still does not work.
Jul 22, 2024
a95bda2
Made some fixes.
philip-paul-mueller Jul 22, 2024
18a8560
If blocking is applied the name of the outer map is now also changed.
philip-paul-mueller Jul 22, 2024
5d979c9
Implemented the possibility to also set the launch bound stuff.
philip-paul-mueller Jul 23, 2024
bcd63d3
Fixed a bug in the auto omptimizer.
philip-paul-mueller Jul 23, 2024
c165f9f
Restructured and cleaned up the auto omptimizer routine.
philip-paul-mueller Jul 26, 2024
316ba9c
Fixed a bug in the `get_map_variable()` function.
philip-paul-mueller Jul 26, 2024
a796766
First batch of stuff for review.
philip-paul-mueller Jul 26, 2024
882ad44
Also checked the map fusion helper stuff.
philip-paul-mueller Jul 26, 2024
a0bf263
Made the reuse of transients optional and disabled it.
philip-paul-mueller Jul 28, 2024
37392fd
First PR candidate for the optimization pipeline.
philip-paul-mueller Jul 29, 2024
5c92c76
Made a small fix in the test function if teh intermnediate was correct.
philip-paul-mueller Jul 29, 2024
f4f5ae5
Added the first series of tests for teh serial map fusion.
philip-paul-mueller Jul 29, 2024
6d12757
Made some small modifications to the map fusion test.
philip-paul-mueller Jul 29, 2024
e590a07
Added a test for the blocking.
philip-paul-mueller Jul 29, 2024
7e99d98
Addressed Edoardo's comments.
philip-paul-mueller Jul 29, 2024
bd35c6d
Forgot to apply some of Edoardo's suggestions.
philip-paul-mueller Jul 29, 2024
0da8ae2
Added myself to the list of authors.
philip-paul-mueller Jul 29, 2024
f978ef7
Merge remote-tracking branch 'gt4py/main' into dace-fieldview-transfo…
philip-paul-mueller Jul 29, 2024
888fb55
Added an utility module.
philip-paul-mueller Jul 29, 2024
0767d6f
Made it possible to extend the applicability of teh map promotion tra…
philip-paul-mueller Jul 29, 2024
c396200
Added a test for the map promotion.
philip-paul-mueller Jul 29, 2024
8e471f1
Reorganized the tests.
philip-paul-mueller Jul 30, 2024
28fcb84
Modified teh first step of teh auto optimizer.
philip-paul-mueller Jul 30, 2024
e8829c6
Added a test to ensure that fusion does not skrew up with indirect ac…
philip-paul-mueller Jul 30, 2024
9373629
Added a todo for a test.
philip-paul-mueller Jul 30, 2024
63a5112
Addressed Edoardo's comment.
philip-paul-mueller Jul 30, 2024
5a2c12c
Added the possibility to controll the iteration order also from teh o…
philip-paul-mueller Jul 31, 2024
bcfbd68
Clarified some buggy behaviour inside the GPU transformation function.
philip-paul-mueller Jul 31, 2024
03f4b1a
Inside a Map there can not be a library node for fusion.
philip-paul-mueller Jul 31, 2024
fd2366f
Applied Edoardo's comments.
philip-paul-mueller Jul 31, 2024
5ed2a8f
Applied another change.
philip-paul-mueller Jul 31, 2024
368c8ad
Merge remote-tracking branch 'gt4py/main' into dace-fieldview-transfo…
philip-paul-mueller Jul 31, 2024
dbc3874
Merge remote-tracking branch 'gt4py/main' into dace-fieldview-transfo…
Aug 2, 2024
8e97cd6
Removed stray symlink.
Aug 2, 2024
390f02b
Merge remote-tracking branch 'gt4py/main' into dace-fieldview-transfo…
philip-paul-mueller Aug 22, 2024
f2b980f
Renamed the file containing the serial map fusion.
philip-paul-mueller Aug 22, 2024
ff4529a
Implemented a first version of the parallel map fusion transformation.
philip-paul-mueller Aug 22, 2024
17af66b
Added tests for the parallel map fusion.
philip-paul-mueller Aug 22, 2024
354f55d
Auto optimizer now uses parallel map fusion too.
philip-paul-mueller Aug 22, 2024
946cd82
Updated the licence file.
philip-paul-mueller Aug 22, 2024
093ff96
Added more control to when to merge.
philip-paul-mueller Aug 22, 2024
4dd7b35
Made a small note about the matchin in the parallel map fusion.
philip-paul-mueller Aug 22, 2024
d7acfd9
Made the import naming a bit more consistent, it is still that great …
philip-paul-mueller Aug 22, 2024
ffbab72
Applied some changing in the naming.
philip-paul-mueller Aug 22, 2024
d45af99
The CI failed and I have no clue why, let's try again.
philip-paul-mueller Aug 23, 2024
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
1 change: 1 addition & 0 deletions AUTHORS.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
- Madonna, Alberto. ETH Zurich - CSCS
- Mariotti, Kean. ETH Zurich - CSCS
- Müller, Christoph. MeteoSwiss
- Müller, Philip. ETH Zurich - CSCS
- Osuna, Carlos. MeteoSwiss
- Paone, Edoardo. ETH Zurich - CSCS
- Röthlin, Matthias. MeteoSwiss
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
# GT4Py - GridTools Framework
#
# Copyright (c) 2014-2024, ETH Zurich
# All rights reserved.
#
# Please, refer to the LICENSE file in the root directory.
# SPDX-License-Identifier: BSD-3-Clause

"""Transformation and optimization pipeline for the DaCe backend in GT4Py.

Please also see [this HackMD document](https://hackmd.io/@gridtools/rklwk4OIR#Requirements-on-SDFG)
that explains the general structure and requirements on the SDFG.
"""

from .auto_opt import dace_auto_optimize, gt_auto_optimize, gt_set_iteration_order, gt_simplify
from .gpu_utils import (
GPUSetBlockSize,
SerialMapPromoterGPU,
gt_gpu_transformation,
gt_set_gpu_blocksize,
)
from .k_blocking import KBlocking
from .map_fusion_parallel import ParallelMapFusion
from .map_fusion_serial import SerialMapFusion
from .map_orderer import MapIterationOrder
from .map_promoter import SerialMapPromoter


__all__ = [
"GPUSetBlockSize",
"KBlocking",
"MapIterationOrder",
"SerialMapFusion",
"SerialMapPromoter",
"SerialMapPromoterGPU",
"ParallelMapFusion",
"dace_auto_optimize",
"gt_auto_optimize",
"gt_gpu_transformation",
"gt_set_iteration_order",
"gt_set_gpu_blocksize",
"gt_simplify",
]
Loading
Loading