diff options
| author | Gangzheng Tong <tonggangzheng@gmail.com> | 2025-06-25 20:09:57 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-06-26 03:09:57 +0000 |
| commit | 5afebcf02748384471a98858eedb685024f7f854 (patch) | |
| tree | b423667a22ebc05b283bbdfdb821f3040e044940 /tests/cooperative-vector | |
| parent | ee51fe592747fc66bd0b5757207583198068b5bd (diff) | |
Add matrix operand for OpCooperativeVectorMatrixMulAddNV (#7524)
* Add matrix operand for OpCooperativeVectorMatrixMulAddNV
* update tests to use the supported UINT32 input component type
* Add MatrixCSignedComponentsKHR for coopVecMatMulAddPacked
---------
Co-authored-by: Jay Kwak <82421531+jkwak-work@users.noreply.github.com>
Diffstat (limited to 'tests/cooperative-vector')
| -rw-r--r-- | tests/cooperative-vector/matrix-mul-bias-packed-mut.slang | 2 | ||||
| -rw-r--r-- | tests/cooperative-vector/matrix-mul-bias-packed.slang | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/cooperative-vector/matrix-mul-bias-packed-mut.slang b/tests/cooperative-vector/matrix-mul-bias-packed-mut.slang index b3443e08a..54d3b11dd 100644 --- a/tests/cooperative-vector/matrix-mul-bias-packed-mut.slang +++ b/tests/cooperative-vector/matrix-mul-bias-packed-mut.slang @@ -25,7 +25,7 @@ ByteAddressBuffer bias; [numthreads(1, 1, 1)] void computeMain() { - let vec = coopVecLoad<1, int32_t>(input); + let vec = coopVecLoad<1, uint32_t>(input); var result = CoopVec<int32_t, 4>(8000); result.matMulAddAccumPacked( vec, diff --git a/tests/cooperative-vector/matrix-mul-bias-packed.slang b/tests/cooperative-vector/matrix-mul-bias-packed.slang index a47a82714..e80fb02ac 100644 --- a/tests/cooperative-vector/matrix-mul-bias-packed.slang +++ b/tests/cooperative-vector/matrix-mul-bias-packed.slang @@ -25,7 +25,7 @@ ByteAddressBuffer bias; [numthreads(1, 1, 1)] void computeMain() { - let vec = coopVecLoad<1, int32_t>(input); + let vec = coopVecLoad<1, uint32_t>(input); let result = coopVecMatMulAddPacked<int32_t, 4>( vec, CoopVecComponentType::SignedInt8Packed, |