Skip to content
This repository has been archived by the owner on Nov 27, 2024. It is now read-only.

Commit

Permalink
fix Parloop kernel arg ordering
Browse files Browse the repository at this point in the history
  • Loading branch information
ksagiyam committed Mar 27, 2024
1 parent 0afe7fa commit a2c0d27
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pyop2/parloop.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,7 @@ def _kernel_args_(self):
@property
def map_kernel_args(self):
rmap, cmap = self.maps
return tuple(itertools.chain(*itertools.product(rmap._kernel_args_, cmap._kernel_args_)))
return tuple(itertools.chain(rmap._kernel_args_, cmap._kernel_args_))


@dataclass
Expand All @@ -143,7 +143,7 @@ def _kernel_args_(self):
@property
def map_kernel_args(self):
rmap, cmap = self.maps
return tuple(itertools.chain(*itertools.product(rmap._kernel_args_, cmap._kernel_args_)))
return tuple(itertools.chain(rmap._kernel_args_, cmap._kernel_args_))


@dataclass
Expand Down

0 comments on commit a2c0d27

Please sign in to comment.