Skip to content

Commit 9a2dbaf

Browse files
committed
[dace] Extra change
1 parent ade5900 commit 9a2dbaf

File tree

1 file changed

+2
-0
lines changed
  • src/gt4py/next/program_processors/runners/dace_iterator

1 file changed

+2
-0
lines changed

src/gt4py/next/program_processors/runners/dace_iterator/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -338,6 +338,8 @@ def build_sdfg_from_itir(
338338
symbols: dict[str, int] = {}
339339
device = dace.DeviceType.GPU if on_gpu else dace.DeviceType.CPU
340340
sdfg = autoopt.auto_optimize(sdfg, device, symbols=symbols, use_gpu_storage=on_gpu)
341+
elif on_gpu:
342+
autoopt.apply_gpu_storage(sdfg)
341343

342344
if on_gpu:
343345
sdfg.apply_gpu_transformations()

0 commit comments

Comments
 (0)