content: import logging
  
  import torch
  from torch_tensorrt.dynamo._settings import CompilationSettings
  from torch_tensorrt.dynamo.lowering.passes.pass_utils import (
      clean_up_graph_after_modifications,
  )
  
  logger = logging.getLogger(__name__)
  
  
  def remove_assert_nodes(
      gm: torch.fx.GraphModule, settings: CompilationSettings
  ) -> torch.fx.GraphModule:
      """Remove assert_scalar ops in the graph"""
      count = 0
      for node in gm.graph.nodes:
          if (
              node.target == torch.ops.aten._assert_scalar.default
              or node.target == torch.ops.aten._assert_tensor_metadata.default
          ):
              gm.graph.erase_node(node)
              count += 1
  
      if count > 0:
          gm = clean_up_graph_after_modifications(gm)
  
      logger.debug(f"Removed {count} assert_scalar nodes:\n{gm.graph}")
  
      return gm
  