facebookresearch / fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.
Apache License 2.0
2k stars 226 forks source link

NotImplementedError during FlopCountAnalysis if submodule decorated by torch.compile #137

Open flishwang opened 1 year ago

flishwang commented 1 year ago

If the model parsed to FlopCountAnalysis contains submodules/functions decorated by torch.compile, the program raises the following errors:

NotImplementedError: Could not run 'aten::_local_scalar_dense' with arguments from the 'Meta' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit .......