Skip to content

Parallel pytato PyOpenCL array context#358

Draft
majosm wants to merge 2 commits into
inducer:mainfrom
majosm:pytato-parallel
Draft

Parallel pytato PyOpenCL array context#358
majosm wants to merge 2 commits into
inducer:mainfrom
majosm:pytato-parallel

Conversation

@majosm
Copy link
Copy Markdown
Collaborator

@majosm majosm commented May 22, 2026

Adds PytatoParallelPyOpenCLArrayContext, which is like PytatoPyOpenCLArrayContext but parallelizes over the device. Based on @kaushikcfd's work in #216. This version generalizes that code to work with fused loops.

The basic procedure is:

  1. Collect instructions that share inames into sets.
  2. Find the "outer" inames (i.e., inames shared between all instructions) in each of those sets.
  3. Parallelize over those.

cc @lukeolson

⚠️ Currently WIP. TODO:

@majosm majosm force-pushed the pytato-parallel branch from e158136 to 66519a6 Compare May 22, 2026 22:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant