Skip to content

Commit 52454c7

Browse files
committed
Add SPV_NV_cooperative_matrix2 and SPV_NV_tensor_addressing
1 parent 252dc2d commit 52454c7

File tree

12 files changed

+1098
-0
lines changed

12 files changed

+1098
-0
lines changed

include/spirv/unified1/spirv.bf

+64
Original file line numberDiff line numberDiff line change
@@ -1204,6 +1204,12 @@ namespace Spv
12041204
AtomicFloat16VectorNV = 5404,
12051205
RayTracingDisplacementMicromapNV = 5409,
12061206
RawAccessChainsNV = 5414,
1207+
CooperativeMatrixReductionsNV = 5430,
1208+
CooperativeMatrixConversionsNV = 5431,
1209+
CooperativeMatrixPerElementOperationsNV = 5432,
1210+
CooperativeMatrixTensorAddressingNV = 5433,
1211+
CooperativeMatrixBlockLoadsNV = 5434,
1212+
TensorAddressingNV = 5439,
12071213
SubgroupShuffleINTEL = 5568,
12081214
SubgroupBufferBlockIOINTEL = 5569,
12091215
SubgroupImageBlockIOINTEL = 5570,
@@ -1437,6 +1443,46 @@ namespace Spv
14371443
Max = 0x7fffffff,
14381444
}
14391445

1446+
[AllowDuplicates, CRepr] public enum CooperativeMatrixReduceShift
1447+
{
1448+
Row = 0,
1449+
Column = 1,
1450+
CooperativeMatrixReduce2x2 = 2,
1451+
Max = 0x7fffffff,
1452+
}
1453+
1454+
[AllowDuplicates, CRepr] public enum CooperativeMatrixReduceMask
1455+
{
1456+
MaskNone = 0,
1457+
Row = 0x00000001,
1458+
Column = 0x00000002,
1459+
CooperativeMatrixReduce2x2 = 0x00000004,
1460+
}
1461+
1462+
[AllowDuplicates, CRepr] public enum TensorClampMode
1463+
{
1464+
Undefined = 0,
1465+
Constant = 1,
1466+
ClampToEdge = 2,
1467+
Repeat = 3,
1468+
RepeatMirrored = 4,
1469+
Max = 0x7fffffff,
1470+
}
1471+
1472+
[AllowDuplicates, CRepr] public enum TensorAddressingOperandsShift
1473+
{
1474+
TensorView = 0,
1475+
DecodeFunc = 1,
1476+
Max = 0x7fffffff,
1477+
}
1478+
1479+
[AllowDuplicates, CRepr] public enum TensorAddressingOperandsMask
1480+
{
1481+
MaskNone = 0,
1482+
TensorView = 0x00000001,
1483+
DecodeFunc = 0x00000002,
1484+
}
1485+
14401486
[AllowDuplicates, CRepr] public enum InitializationModeQualifier
14411487
{
14421488
InitOnDeviceReprogramINTEL = 0,
@@ -1958,6 +2004,7 @@ namespace Spv
19582004
OpReorderThreadWithHintNV = 5280,
19592005
OpTypeHitObjectNV = 5281,
19602006
OpImageSampleFootprintNV = 5283,
2007+
OpCooperativeMatrixConvertNV = 5293,
19612008
OpEmitMeshTasksEXT = 5294,
19622009
OpSetMeshOutputsEXT = 5295,
19632010
OpGroupNonUniformPartitionNV = 5296,
@@ -1982,9 +2029,26 @@ namespace Spv
19822029
OpCooperativeMatrixLengthNV = 5362,
19832030
OpBeginInvocationInterlockEXT = 5364,
19842031
OpEndInvocationInterlockEXT = 5365,
2032+
OpCooperativeMatrixReduceNV = 5366,
2033+
OpCooperativeMatrixLoadTensorNV = 5367,
2034+
OpCooperativeMatrixStoreTensorNV = 5368,
2035+
OpCooperativeMatrixPerElementOpNV = 5369,
2036+
OpTypeTensorLayoutNV = 5370,
2037+
OpTypeTensorViewNV = 5371,
2038+
OpCreateTensorLayoutNV = 5372,
2039+
OpTensorLayoutSetDimensionNV = 5373,
2040+
OpTensorLayoutSetStrideNV = 5374,
2041+
OpTensorLayoutSliceNV = 5375,
2042+
OpTensorLayoutSetClampValueNV = 5376,
2043+
OpCreateTensorViewNV = 5377,
2044+
OpTensorViewSetDimensionNV = 5378,
2045+
OpTensorViewSetStrideNV = 5379,
19852046
OpDemoteToHelperInvocation = 5380,
19862047
OpDemoteToHelperInvocationEXT = 5380,
19872048
OpIsHelperInvocationEXT = 5381,
2049+
OpTensorViewSetClipNV = 5382,
2050+
OpTensorLayoutSetBlockSizeNV = 5384,
2051+
OpCooperativeMatrixTransposeNV = 5390,
19882052
OpConvertUToImageNV = 5391,
19892053
OpConvertUToSamplerNV = 5392,
19902054
OpConvertImageToUNV = 5393,

0 commit comments

Comments
 (0)