pytorch - 💡(How to fix) Fix Does torch.compile support zero dimensions?

Hi,

I am compiling a model with

torch.compile(compile_mode='default', compile_dynamic=None, compile_fullgraph=True,compile_backend='inductor')

But it seems to fail with all of a sudden with [rank0]: RuntimeError: PassManager::run failed error.

I tried enabling torch compile logs but apart from spamming 1M lines of text, I don't see anything more relevant as an error.

The only thing I can think about is that I have pieces of my model I can disable from flags. When that happens, they produce tensors of dimensions (Batch, 0) (on purpose, so that the code is a bit easier to follow).

Could this be a problem for triton?

cc @chauhang @penguinwu @voznesenskym @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @kadeng @muchulee8 @amjames @aakhundov @coconutruben @jataylo