From 52f4584969ea2b5271fd7713a3ba92e30ee886b8 Mon Sep 17 00:00:00 2001 From: Scott Sievert Date: Sun, 3 Jul 2016 11:07:09 -0500 Subject: [PATCH] Provide gpuarray.dot --- pycuda/gpuarray.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pycuda/gpuarray.py b/pycuda/gpuarray.py index 896b2fca..3f145f7a 100644 --- a/pycuda/gpuarray.py +++ b/pycuda/gpuarray.py @@ -244,6 +244,9 @@ def set(self, ary, async=False, stream=None): def set_async(self, ary, stream=None): return self.set(ary, async=True, stream=stream) + + def dot(self, x, stream=None, allocator=None): + return dot(self, x, stream=stream, allocator=allocator) def get(self, ary=None, pagelocked=False, async=False, stream=None): if ary is None: