old_contents: #!/usr/bin/env python3
  from run import run
  
  
  def main():
      params = {}
      params["framework"] = "pytorch"
      params["problem"] = "bert_custom"
      params["mode"] = "training"
      params["gpus"] = "0"
      for batch_size in range(1, 256):
          params["batch_size"] = batch_size
          prob_size = params["batch_size"] * 4
          params["problem_size"] = f"{prob_size}, 128"
          print(batch_size)
          run(params)
          params.pop("gpus")
          params["backend"] = "DNNL"
          print(batch_size)
          run(params)
  
  
  if __name__ == "__main__":
      main()
  
new_contents: #!/usr/bin/env python3
  from run import run
  
  
  def main():
      params = {}
      params["framework"] = "pytorch"
      params["problem"] = "bert_custom"
      params["mode"] = "training"
      for batch_size in range(1, 256):
          params["gpus"] = "0"
          params["backend"] = "native"
          params["batch_size"] = batch_size
          prob_size = params["batch_size"] * 4
          params["problem_size"] = f"{prob_size}, 128"
          print(batch_size)
          run(params)
          params.pop("gpus")
          params["backend"] = "DNNL"
          print(batch_size)
          run(params)
  
  
  if __name__ == "__main__":
      main()
  
subject: fix bert