From 1ba3e93e62490e744c2d6581ad8d759d988a3a82 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 16 Jul 2013 17:26:54 -0400
Subject: [PATCH] fix_parameter -> fix_parameters

---
 doc/reference.rst  | 2 +-
 loopy/__init__.py  | 9 ++++++++-
 test/test_loopy.py | 5 ++---
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/doc/reference.rst b/doc/reference.rst
index 81d1acd14..4239a498a 100644
--- a/doc/reference.rst
+++ b/doc/reference.rst
@@ -257,7 +257,7 @@ Wrangling inames
 Dealing with Parameters
 ^^^^^^^^^^^^^^^^^^^^^^^
 
-.. autofunction:: fix_parameter
+.. autofunction:: fix_parameters
 
 Dealing with Substitution Rules
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
diff --git a/loopy/__init__.py b/loopy/__init__.py
index c2289f478..6d51293e4 100644
--- a/loopy/__init__.py
+++ b/loopy/__init__.py
@@ -1173,7 +1173,7 @@ def split_reduction_outward(kernel, inames, within=None):
 
 # {{{ fix_parameter
 
-def fix_parameter(kernel, name, value):
+def _fix_parameter(kernel, name, value):
     def process_set(s):
         var_dict = s.get_var_dict()
 
@@ -1232,6 +1232,13 @@ def fix_parameter(kernel, name, value):
                 assumptions=process_set(kernel.assumptions),
                 ))
 
+
+def fix_parameters(kernel, **value_dict):
+    for name, value in value_dict.iteritems():
+        kernel = _fix_parameter(kernel, name, value)
+
+    return kernel
+
 # }}}
 
 
diff --git a/test/test_loopy.py b/test/test_loopy.py
index c6489bf57..d425bf94f 100644
--- a/test/test_loopy.py
+++ b/test/test_loopy.py
@@ -1351,8 +1351,7 @@ def test_rob_stroud_bernstein(ctx_factory):
             assumptions="deg>=0 and nels>=1"
             )
 
-    knl = lp.fix_parameter(knl, "nqp1d", 7)
-    knl = lp.fix_parameter(knl, "deg", 4)
+    knl = lp.fix_parameters(knl, nqp1d=7, deg=4)
     knl = lp.split_iname(knl, "el", 16, inner_tag="l.0")
     knl = lp.split_iname(knl, "el_outer", 2, outer_tag="g.0", inner_tag="ilp",
             slabs=(0, 1))
@@ -1380,7 +1379,7 @@ def test_vector_types(ctx_factory, vec_len):
                 "..."
                 ])
 
-    knl = lp.fix_parameter(knl, "vec_len", vec_len)
+    knl = lp.fix_parameters(knl, vec_len=vec_len)
 
     ref_knl = knl
 
-- 
GitLab