From 35411cdc4cda60342534b8a3c88006365e70b02d Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 16 May 2021 13:34:55 -0500
Subject: [PATCH] zeros, zeros_like: minor speedups
---
pyopencl/array.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/pyopencl/array.py b/pyopencl/array.py
index 40187cb8..e05ba222 100644
--- a/pyopencl/array.py
+++ b/pyopencl/array.py
@@ -2138,8 +2138,9 @@ def zeros(queue, shape, dtype, order="C", allocator=None):
*context* argument was deprecated.
"""
- result = Array(queue, shape, dtype,
- order=order, allocator=allocator)
+ result = Array(None, shape, dtype,
+ order=order, allocator=allocator,
+ _context=queue.context, _queue=queue)
result._zero_fill()
return result
@@ -2158,7 +2159,7 @@ def zeros_like(ary):
as *other_ary*.
"""
- result = empty_like(ary)
+ result = ary._new_like_me()
result._zero_fill()
return result
--
GitLab