Skip to content

Commit

Permalink
Minor format change, getting ready to merge.
Browse files Browse the repository at this point in the history
  • Loading branch information
liuliu committed Aug 15, 2024
1 parent fc9e662 commit 9cd5e03
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions lib/nnc/mfa/v2/GEMMDescriptor.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -114,10 +114,10 @@ std::pair<GEMMKernelDescriptor, PipelineValue<GEMMKernel> *> GEMMDescriptor::fin
bool batched = this->batchDimension > 1;
constants->setConstantValue(&batched, MTL::DataTypeBool, 11);
simd::uint4 batchStrides = this->batchStrides.value_or(simd::uint4(0));
auto batchStrideA = batchStrides[0];
auto batchStrideB = batchStrides[1];
auto batchStrideC = batchStrides[2];
auto batchStrideBias = batchStrides[3];
auto batchStrideA = batchStrides[0];
auto batchStrideB = batchStrides[1];
auto batchStrideC = batchStrides[2];
auto batchStrideBias = batchStrides[3];
constants->setConstantValue(&batchStrideA, MTL::DataTypeUInt, 15);
constants->setConstantValue(&batchStrideB, MTL::DataTypeUInt, 16);
constants->setConstantValue(&batchStrideC, MTL::DataTypeUInt, 17);
Expand Down

0 comments on commit 9cd5e03

Please sign in to comment.