[GitHub] [tvm] vinx13 commented on a diff in pull request #14770: [Vulkan] Add `VK_NV_cooperative_matrix` support

["vinx13 (via GitHub)" <gi...@apache.org>]

vinx13 commented on code in PR #14770:
URL: https://github.com/apache/tvm/pull/14770#discussion_r1194650947


##########
src/target/spirv/codegen_spirv.cc:
##########
@@ -561,6 +630,55 @@ void CodeGenSPIRV::VisitStmt_(const ForNode* op) {
 
   // Loop head
   builder_->StartLabel(head_label);
+
+  // In a normal matmul, the update semantics c += a * b is implemented by load and store to the

Review Comment:
   Being able to reuse existing schedule and auto tensorization for WMMA is a big advantage. I'm wondering if we can reuse wmma intrinsics since we can still use fragment index to identify matrix (if we want to scalarize it)



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org