From 01805a14891e53196c4f9d877176e2d6f79ce51e Mon Sep 17 00:00:00 2001 From: Matt Wala Date: Wed, 22 Jul 2020 13:25:05 -0500 Subject: [PATCH] Change temporary variable address space to auto --- pytato/codegen.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytato/codegen.py b/pytato/codegen.py index 86a84f2..54c1503 100644 --- a/pytato/codegen.py +++ b/pytato/codegen.py @@ -377,7 +377,7 @@ class CodeGenMapper(pytato.transform.Mapper): out = lp.TemporaryVariable(out_name, dtype=expr.dtype, shape=expr.shape, - address_space=lp.AddressSpace.GLOBAL) + address_space=lp.auto) inames = [] for j in range(expr.ndim - 1): @@ -641,7 +641,7 @@ def add_store(name: str, expr: Array, result: ImplementedResult, tvar = lp.TemporaryVariable(name, dtype=expr.dtype, shape=expr.shape, - address_space=lp.AddressSpace.GLOBAL) + address_space=lp.auto) temporary_variables = kernel.temporary_variables.copy() temporary_variables[name] = tvar kernel = kernel.copy(temporary_variables=temporary_variables, -- GitLab