NVlabs / timeloop

Timeloop performs modeling, mapping and code-generation for tensor algebra workloads on various accelerator architectures.
https://timeloop.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
325 stars 101 forks source link

Workload propagation #270

Closed angshuman-parashar closed 3 months ago

angshuman-parashar commented 3 months ago

The workload object in the master branch is a global object that represents a single unmapped Einsum. This set of changes removes the global object and propagates the workload object throughout the object/method hierarchy. This is safer, and sets the stage for modeling multi-Einsum workloads.