[Tracer] Remove SelfAttention renaming

[chhzh123]

Description

This PR fixes #13 and #30 by removing the legacy code of renaming self attention module for HuggingFace models. As now we only trace the module when needed and also replace the attention module with our own module, the naming issue probably may not be a problem. We also do not refer to self_m in our example schedules, so I remove the renaming part from the tracer, and only prompt a warning when the tracer meets those self modules.

Checklist

• PR's title starts with a category (e.g. [Bugfix], [Model], [Tutorial], etc)
• Changes are complete (i.e. I finished coding on this PR)
• All changes have test coverage
• Code is well-documented

cc @szhengac @comaniac

[comaniac]

This is just a warning. What happen if we keep running?

[chhzh123]

I tested the traced module and it can execute normally. If we only call the SelfAttention module with the self prefix, it seems to work. I forget why at the first place I met the naming error, which was probably related to the previous implementation of the tracer.

class Model(nn.Module):
  def __init__(self):
    self.self = ...

  def forward(self, x):
    x = self.self(x)
    ...

[comaniac]

Ok got it.

[comaniac]

Thanks @chhzh123