From e5fc8d7deafffc2bb78e542a8c71f02fdece02de Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 24 Aug 2012 00:30:15 -0400
Subject: [PATCH] Fix initialization of vectors in reduction.
---
loopy/codegen/expression.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/loopy/codegen/expression.py b/loopy/codegen/expression.py
index 870843aaa..a0f430545 100644
--- a/loopy/codegen/expression.py
+++ b/loopy/codegen/expression.py
@@ -134,6 +134,10 @@ def dtype_to_type_context(dtype):
return 'd'
if dtype in [np.float32, np.complex64]:
return 'f'
+ from pyopencl.array import vec
+ if dtype in vec.types.values():
+ return dtype_to_type_context(dtype.fields["x"][0])
+
return None
--
GitLab