Skip to content

Commit

Permalink
[js/webgpu] Fix mha name (microsoft#20860)
Browse files Browse the repository at this point in the history
### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
  • Loading branch information
axinging authored May 30, 2024
1 parent 228713f commit 25ac653
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion js/web/lib/wasm/jsep/webgpu/op-resolve-rules.ts
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import {instanceNorm} from './ops/instance-norm';
import {layerNorm} from './ops/layer-norm';
import {matMul} from './ops/matmul';
import {matMulNBits, parseMatMulNBitsAttributes} from './ops/matmulnbits';
import {multiHeadAttention, parseMultiHeadAttentionAttributes} from './ops/multihead-attentiion';
import {multiHeadAttention, parseMultiHeadAttentionAttributes} from './ops/multihead-attention';
import {pad} from './ops/pad';
import * as pool from './ops/pool';
import {range} from './ops/range';
Expand Down
2 changes: 1 addition & 1 deletion js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ import {ComputeContext, ProgramInfo, ProgramInputTensorInfoDependency, ProgramUn

import {applyAttention, AttentionAttrs, AttentionMaskType, AttentionParameters, AttentionQkvFormat} from './attention';
import {createTensorShapeVariables, inputVariable, outputVariable, ShaderHelper, UniformsArrayType} from './common';
import {maybeTransposeToBNSHAndAddBias} from './multihead-attentiion';
import {maybeTransposeToBNSHAndAddBias} from './multihead-attention';
import {createTileProgramInfo} from './tile';
import {createTransposeProgramInfo, TransposeAttributes} from './transpose';

Expand Down

0 comments on commit 25ac653

Please sign in to comment.