From f1e6b6d19f7b1491a74c2975743709a6f46dee3d Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Tue, 16 Jul 2013 23:41:07 -0400 Subject: [PATCH] Fix vector access --- loopy/codegen/expression.py | 8 +------- test/test_loopy.py | 2 +- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/loopy/codegen/expression.py b/loopy/codegen/expression.py index f1719a86b..61fd776f1 100644 --- a/loopy/codegen/expression.py +++ b/loopy/codegen/expression.py @@ -237,17 +237,11 @@ def dtype_to_type_context(dtype): return None -VEC_AXES = "xyzw" - - def get_opencl_vec_member(idx): if idx is None: return idx - if idx < len(VEC_AXES): - return VEC_AXES[idx] - else: - return "s%d" % idx + return "s%s" % hex(idx)[2:] class SeenFunction(Record): diff --git a/test/test_loopy.py b/test/test_loopy.py index b3ec3f8f1..7de8d13bd 100644 --- a/test/test_loopy.py +++ b/test/test_loopy.py @@ -1439,7 +1439,7 @@ def test_rob_stroud_bernstein(ctx_factory): )) -@pytest.mark.parametrize("vec_len", [2, 3, 4, 8]) +@pytest.mark.parametrize("vec_len", [2, 3, 4, 8, 16]) def test_vector_types(ctx_factory, vec_len): ctx = cl.create_some_context() -- GitLab