From ad8f708ff0e213ecb53807ca2443dacea06eea17 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 19 Feb 2014 15:01:38 -0600
Subject: [PATCH] Fix pickling of ArgExtFunction
---
loopy/library/reduction.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/loopy/library/reduction.py b/loopy/library/reduction.py
index 9071cbea8..fa382cb45 100644
--- a/loopy/library/reduction.py
+++ b/loopy/library/reduction.py
@@ -162,12 +162,17 @@ class ArgMinReductionOperation(_ArgExtremumReductionOperation):
class ArgExtFunction(FunctionIdentifier):
+ init_arg_names = ("reduction_op", "scalar_dtype", "name", "inames")
+
def __init__(self, reduction_op, scalar_dtype, name, inames):
self.reduction_op = reduction_op
self.scalar_dtype = scalar_dtype
self.name = name
self.inames = inames
+ def __getinitargs(self):
+ return (self.reduction_op, self.scalar_dtype, self.name, self.inames)
+
def get_argext_preamble(func_id):
op = func_id.reduction_op
--
GitLab