Skip to content

Commit 6504469

Browse files
kaushikcfdinducer
authored andcommitted
Use 'optimal' strategy for einsum contraction path.
1 parent b19f696 commit 6504469

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

arraycontext/impl/numpy/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def tag_axis(self,
197197
return array
198198

199199
def einsum(self, spec, *args, arg_names=None, tagged=()):
200-
return np.einsum(spec, *args)
200+
return np.einsum(spec, *args, optimize="optimal")
201201

202202
@property
203203
def permits_inplace_modification(self):

0 commit comments

Comments
 (0)