Support 2D pointwise scheduler with autotuning - Githubissues

NVIDIA / Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

Other

271 stars 53 forks source link

Support 2D pointwise scheduler with autotuning #3407

Open rdspring1 opened 1 week ago

rdspring1 commented 1 week ago

Refactored autotune_pointwise.py to support 2D pointwise scheduler
Added support for Gelu-Bias, Silu-Mul, Bcast-Add, Mul fusions
Use ensemble.RandomForestRegressor instead of tree.DecisionTreeRegressor

rdspring1 commented 6 days ago

!build