From cf241efcb0633e7f790f41574bea3780c141f3e0 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Tue, 23 Jul 2024 14:08:14 +0200
Subject: [PATCH] two more minor API fixes
---
src/pystencils/datahandling/serial_datahandling.py | 4 ++--
src/pystencils/runhelper/db.py | 2 +-
tests/runtime/test_datahandling.py | 3 ---
3 files changed, 3 insertions(+), 6 deletions(-)
diff --git a/src/pystencils/datahandling/serial_datahandling.py b/src/pystencils/datahandling/serial_datahandling.py
index 4eb341df5..ba705f4b9 100644
--- a/src/pystencils/datahandling/serial_datahandling.py
+++ b/src/pystencils/datahandling/serial_datahandling.py
@@ -344,11 +344,11 @@ class SerialDataHandling(DataHandling):
if target == Target.CPU:
def result_functor():
for arr_name, func in zip(names, result):
- func(pdfs=self.cpu_arrays[arr_name])
+ func(self.cpu_arrays[arr_name])
else:
def result_functor():
for arr_name, func in zip(names, result):
- func(pdfs=self.gpu_arrays[arr_name])
+ func(self.gpu_arrays[arr_name])
return result_functor
diff --git a/src/pystencils/runhelper/db.py b/src/pystencils/runhelper/db.py
index 466b9dc14..dd413a5e4 100644
--- a/src/pystencils/runhelper/db.py
+++ b/src/pystencils/runhelper/db.py
@@ -8,7 +8,7 @@ import six
from blitzdb.backends.file.backend import serializer_classes
from blitzdb.backends.file.utils import JsonEncoder
-from pystencils.cpu.cpujit import get_compiler_config
+from pystencils.backend.jit.legacy_cpu import get_compiler_config
from pystencils import CreateKernelConfig, Target, Field
import json
diff --git a/tests/runtime/test_datahandling.py b/tests/runtime/test_datahandling.py
index 90c21db18..29e639c88 100644
--- a/tests/runtime/test_datahandling.py
+++ b/tests/runtime/test_datahandling.py
@@ -408,6 +408,3 @@ def test_array_handler(device_number):
assert cpu_array2.base is not None
assert gpu_array2.base is not None
assert gpu_array2.strides == cpu_array2.strides
-
-
-test_access_and_gather()
--
GitLab