GauS: Differentiable Scheduling Optimization via Gaussian Reparameterization
Researchers have introduced GauS, a novel differentiable framework for optimizing operator scheduling in software compilation and hardware synthesis. Unlike previous methods that used categorical distributions, GauS employs Gaussian distributions to better capture the ordinal nature of time and significantly reduce the optimization space. This approach is flexible for various objectives and constraints, offering the first differentiable formulation for complex pipelined scheduling problems. Evaluations on benchmarks show GauS achieving Pareto-optimal results, leveraging modern parallel computing devices like GPUs. AI