Precompile all kernels (no jit)

[YLGH]

Hi, thanks for the amazing library!

Just wondering, is there a way to disable jit by precompiling all the necessary shapes? Or is it recommended to just some "warmup" script to compile all the kernels? What's a good list of M,N,K values to do so?

[jason-huang03]

Not possible, the N and K are template arguments.

[LyricZhao]

For your own model (if the shapes are fixed), you can simply modify the test scripts to test all the shapes your model used. And later for model training, make sure that you are using the same cache dir (DG_CACHE_DIR).