code: def forward(
      self, x, residual=None, prenorm=False, residual_in_fp32=False
  ):
      return rms_norm_fn(
          x,
          self.weight,
          self.bias,
          residual=residual,
          eps=self.eps,
          dropout_p=self.drop.p
          if self.drop is not None and self.training
          else 0.0,
          prenorm=prenorm,
          residual_in_fp32=residual_in_fp32,
      )
  
imports: import math
  import torch
  import torch.nn.functional
  from torch.cuda.amp import custom_fwd, custom_bwd
  import triton
  import triton.language
label_desc: Use image node_id 3 for calling the RMSNorm obj's underlying member method code with example usage: obj.forward(x, residual, prenorm, residual_in_fp32) and returns: rms_norm_fn