Would it be easy to extend your gemm schedule into a schedule for BatchMatMul? 
That would help round out the TensorCore story for matrix multiplication.

-- 
You are receiving this because you are subscribed to this thread.
Reply to this email directly or view it on GitHub:
https://github.com/dmlc/tvm/issues/4052#issuecomment-538852791

Reply via email to