From 25ac65375cb590043a6e7243eb251fd193b5413c Mon Sep 17 00:00:00 2001 From: Xu Xing Date: Thu, 30 May 2024 15:01:06 +0800 Subject: [PATCH] [js/webgpu] Fix mha name (#20860) ### Description ### Motivation and Context --- js/web/lib/wasm/jsep/webgpu/op-resolve-rules.ts | 2 +- js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts | 2 +- .../ops/{multihead-attentiion.ts => multihead-attention.ts} | 0 3 files changed, 2 insertions(+), 2 deletions(-) rename js/web/lib/wasm/jsep/webgpu/ops/{multihead-attentiion.ts => multihead-attention.ts} (100%) diff --git a/js/web/lib/wasm/jsep/webgpu/op-resolve-rules.ts b/js/web/lib/wasm/jsep/webgpu/op-resolve-rules.ts index 78e4871dcec8e..2d2f345d0c273 100644 --- a/js/web/lib/wasm/jsep/webgpu/op-resolve-rules.ts +++ b/js/web/lib/wasm/jsep/webgpu/op-resolve-rules.ts @@ -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'; diff --git a/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts b/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts index d038205910577..0558d1caf76a6 100644 --- a/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts +++ b/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts @@ -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'; diff --git a/js/web/lib/wasm/jsep/webgpu/ops/multihead-attentiion.ts b/js/web/lib/wasm/jsep/webgpu/ops/multihead-attention.ts similarity index 100% rename from js/web/lib/wasm/jsep/webgpu/ops/multihead-attentiion.ts rename to js/web/lib/wasm/jsep/webgpu/ops/multihead-attention.ts