code: def benchmark_cublas(dims: MatmulSize) -> MatmulTiming:
      """Measure cublas performance."""
      a = torch.randn(
          dims.M, dims.K, device="cuda", dtype=torch.bfloat16
      )
      b = torch.randn(
          dims.K, dims.N, device="cuda", dtype=torch.bfloat16
      )
      run_matmul = lambda: torch.matmul(a, b)
      percentiles = triton.testing.do_bench(
          run_matmul,
          warmup=0,
          rep=300,
          quantiles=(0.001, 0.1, 0.5, 0.9),
      )
      min_ms = percentiles[0]
      return min_ms
  
imports: import enum
  import itertools
  import logging
  import math
  import typing
  import torch
  import tqdm
  import triton
  import triton.language
label_desc: Use image node_id 10 for calling a global function with example usage: benchmark_cublas(dims) and returns: min_ms