From be32e3aaa57a81479050609a36625149b2549c05 Mon Sep 17 00:00:00 2001
From: Alex Fikl <alexfikl@gmail.com>
Date: Thu, 8 Mar 2018 21:32:53 -0600
Subject: [PATCH] qbx: remove dtype from result
---
sumpy/p2p.py | 10 +++++-----
sumpy/qbx.py | 12 ++++++------
2 files changed, 11 insertions(+), 11 deletions(-)
diff --git a/sumpy/p2p.py b/sumpy/p2p.py
index ce9fc2c8..84c8f390 100644
--- a/sumpy/p2p.py
+++ b/sumpy/p2p.py
@@ -358,10 +358,10 @@ class P2PMatrixBlockGenerator(P2PBase):
lang_version=MOST_RECENT_LANGUAGE_VERSION)
loopy_knl = lp.add_dtypes(loopy_knl, dict(
- nsources=np.int64,
- ntargets=np.int64,
- ntgtindices=np.int64,
- nsrcindices=np.int64))
+ nsources=np.int32,
+ ntargets=np.int32,
+ ntgtindices=np.int32,
+ nsrcindices=np.int32))
loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr")
for knl in self.kernels:
@@ -479,7 +479,7 @@ class P2PFromCSR(P2PBase):
lang_version=MOST_RECENT_LANGUAGE_VERSION)
loopy_knl = lp.add_dtypes(loopy_knl,
- dict(nsources=np.int64, ntargets=np.int64))
+ dict(nsources=np.int32, ntargets=np.int32))
loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr")
loopy_knl = lp.tag_array_axes(loopy_knl, "targets", "sep,C")
diff --git a/sumpy/qbx.py b/sumpy/qbx.py
index f0c35174..5d4e1105 100644
--- a/sumpy/qbx.py
+++ b/sumpy/qbx.py
@@ -202,8 +202,8 @@ class LayerPotential(LayerPotentialBase):
None, shape="nsources", order="C")
for i in range(self.strength_count)] +
[lp.GlobalArg("result_%d" % i,
- dtype, shape="ntargets", order="C")
- for i, dtype in enumerate(self.value_dtypes)])
+ None, shape="ntargets", order="C")
+ for i, _ in enumerate(self.value_dtypes)])
loopy_knl = lp.make_kernel([
"{[itgt]: 0 <= itgt < ntargets}",
@@ -386,10 +386,10 @@ class LayerPotentialMatrixBlockGenerator(LayerPotentialBase):
lang_version=MOST_RECENT_LANGUAGE_VERSION)
loopy_knl = lp.add_dtypes(loopy_knl, dict(
- nsources=np.int64,
- ntargets=np.int64,
- ntgtindices=np.int64,
- nsrcindices=np.int64))
+ nsources=np.int32,
+ ntargets=np.int32,
+ ntgtindices=np.int32,
+ nsrcindices=np.int32))
loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr")
for expn in self.expansions:
--
GitLab