Skip to content

Commit

Permalink
[FRONTEND] Refactor jit.py. (#2556)
Browse files Browse the repository at this point in the history
[FRONTEND] Refactor jit.py.

The goal is to simplify the code and make it more flexible before we
change the kernel launch syntax to
`kernel[grid, compiler_flags(...)](...)`.

The main changes here are:

 - Get rid of the eval'ed code in make_launcher.  We can do everything
   using bind().
 - Add KernelParam and KernelArg classes, letting us get rid of the
   parallel arrays/dicts indexed by parameter index.
 - Get rid of duplicated kernel launch code in the cache-hit/cache-miss
   branches.
  • Loading branch information
jlebar committed Oct 30, 2023
1 parent f88b01f commit 12f9062
Showing 1 changed file with 269 additions and 258 deletions.
Loading

0 comments on commit 12f9062

Please sign in to comment.