hidet-org/hidet

torch.nn.Identity are not supported yet

carl10086 opened this issue · 0 comments

carl10086 commented

Describe the bug
Some errors occured when I compiled dinov2

To Reproduce

model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14')
model = model.to('cuda')
model_opt = torch.compile(model, backend='hidet')

Error information

torch._dynamo.exc.BackendCompilerFailed: hidet_backend raised NotImplementedError: The following modules/functions are not supported by hidet yet:
  torch.nn.Identity

Enviroment

  • OS: Ubuntu 22.04
  • GPU: A100

Additional context
Add any other context about the problem here.