diff --git a/src/gt4py/next/program_processors/runners/gtfn.py b/src/gt4py/next/program_processors/runners/gtfn.py index 396c921e2a..7233e7a893 100644 --- a/src/gt4py/next/program_processors/runners/gtfn.py +++ b/src/gt4py/next/program_processors/runners/gtfn.py @@ -191,5 +191,5 @@ def compilation_hash(otf_closure: stages.ProgramCall) -> int: ) run_gtfn_gpu_cached = otf_compile_executor.OTFBackend( executor=gtfn_gpu_cached_executor, - allocator=next_allocators.StandardCPUFieldBufferAllocator(), + allocator=next_allocators.StandardGPUFieldBufferAllocator(), )