class_docstr: Fused operation which performs following three operations in sequence
  1. Scale the tensor.
  2. Apply the mask.
  3. Perform softmax.
code: def forward(ctx, inputs, mask, scale):
      import scaled_masked_softmax_cuda
  
      scale_t = torch.tensor([scale])
  
      softmax_results = scaled_masked_softmax_cuda.forward(
          inputs, mask, scale_t[0]
      )
      ctx.save_for_backward(softmax_results, scale_t)
      return softmax_results
  
imports: import torch
  import torch.nn
  from megatron.core.transformer.enums import AttnMaskType
label_desc: Use image node_id 1 for calling the ScaledMaskedSoftmax obj's underlying member method code with example usage: obj.forward(ctx, inputs, mask, scale) and returns: softmax_results