code: def reset_cudnn_benchmark():
      """Ensures that the `torch.backends.cudnn.benchmark` setting gets reset before the next test runs."""
      yield
      torch.backends.cudnn.benchmark = False
  
imports: import os
  import pytest
  import torch.distributed
label_desc: Use image node_id 2 for calling a global function with example usage: reset_cudnn_benchmark() without return types