code: def forward(self, x):
      if torch.is_autocast_enabled() and x.dtype != torch.float32:
          with torch.cuda.amp.autocast(enabled=False):
              ret = self.mod.forward(x.to(torch.float32)).to(x.dtype)
      else:
          ret = self.mod.forward(x)
      return ret
  
file_path: NeMo/nemo/utils/cast_utils.py
imports: from contextlib import nullcontext
  import torch
label_desc: Use image node_id 2 for calling the CastToFloat obj's underlying member method code with example usage: obj.forward(x) and returns: ret