19de6250f0c665cfd2bbf6b1f0293dbf39634ba3,tutorials/optimize/opt_matmul_auto_tensorcore.py,,,#,239

Before Change

ctx = tvm.gpu()
if not nvcc.have_tensorcore(ctx.compute_version):
  print("the gpu has no tensorcore, skipping...")
  sys.exit(0)

M, N, L = 512, 32, 512
dtype = "float16"
layout = "NN"

After Change

ctx = tvm.gpu()
if not nvcc.have_tensorcore(ctx.compute_version):
  raise Exception("the gpu has no tensorcore, skipping...")

M, N, L = 512, 32, 512
dtype = "float16"
layout = "NN"

Instances