Specific Demand
Fusor is essentially a JIT ML compiler. The JIT part makes the compute graphs you can generate very flexible, but it makes PGO more difficult.
Instead of performing PGO on the whole compute graph, we could chunk the graph into sections and analyze the performance of those chunks. The individual chunks are more likely to be re-used in future runs of the model.
Implement Suggestion
- Chunk the compute graph into small sections
- Modify the subgraphs with a monte-carlo tree search
- Cache the intermediate results of the tree search candidate for future runs
Specific Demand
Fusor is essentially a JIT ML compiler. The JIT part makes the compute graphs you can generate very flexible, but it makes PGO more difficult.
Instead of performing PGO on the whole compute graph, we could chunk the graph into sections and analyze the performance of those chunks. The individual chunks are more likely to be re-used in future runs of the model.
Implement Suggestion