Skip to content

Parallel pytato PyOpenCL array context#358

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

Parallel pytato PyOpenCL array context#358
majosm wants to merge 3 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 4 times, most recently from 8a7d2a7 to 7543ce8 Compare May 26, 2026 19:16
@majosm majosm force-pushed the pytato-parallel branch from 7543ce8 to c687225 Compare May 27, 2026 19:22
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