code: def main(bit_rate):
      # uncomment for debugging
      # np.random.seed(0)
      batchsize = 10 * 1000
      blocksize = 64
      print(batchsize, blocksize)
      input_data = np.random.rand(batchsize, blocksize).astype(
          np.float32
      )
  
      workspace.FeedBlob("input_data", input_data)
  
      net = core.Net("bench")
      op = core.CreateOperator(
          "FloatToFused" + str(bit_rate) + "BitRowwiseQuantized",
          "input_data",
          "quantized_data",
          engine="GREEDY",
      )
      net.Proto().op.extend([op])
      workspace.GlobalInit(["caffe2", "--caffe2_log_level=0"])
      workspace.CreateNet(net)
      iterations = 10
      workspace.BenchmarkNet(net.Proto().name, 1, iterations, True)
  
      net2 = core.Net("bench2")
      op = core.CreateOperator(
          "FloatToFused" + str(bit_rate) + "BitRowwiseQuantized",
          "input_data",
          "quantized_data",
      )
      net2.Proto().op.extend([op])
  
      workspace.CreateNet(net2)
      workspace.BenchmarkNet(net2.Proto().name, 1, iterations, True)
  
label_desc: Use image node_id 1 for calling a global function with example usage: main(bit_rate) without return types