Toward Compiler World Models: Learning Latent Dynamics for Efficient Tensor Program Search

Haolin Pan, Lianghong Huang, Xvlin Zhou +2

Tensor program optimization is essential for modern machine learning systems, but its search space is enormous. Existing auto-schedulers reduce measurement cost with learned cost m…