File tree 2 files changed +6
-2
lines changed
js/web/lib/wasm/jsep/webgpu
2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -436,7 +436,11 @@ export const createMatMulNBitsBlockSize32ProgramInfo = (
436
436
437
437
export const matMulNBits = ( context : ComputeContext , attributes : MatMulNBitsAttributes ) : void => {
438
438
validateInputs ( context . inputs , attributes ) ;
439
- if ( attributes . blockSize === 32 && context . adapterInfo . isVendor ( 'intel' ) ) {
439
+ if (
440
+ attributes . blockSize === 32 &&
441
+ context . adapterInfo . isVendor ( 'intel' ) &&
442
+ context . adapterInfo . isArchitecture ( 'gen-12lp' )
443
+ ) {
440
444
context . compute ( createMatMulNBitsBlockSize32ProgramInfo ( context . inputs , attributes ) ) ;
441
445
} else {
442
446
context . compute ( createMatMulNBitsProgramInfo ( context . inputs , attributes ) ) ;
Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ export enum GpuDataType {
15
15
}
16
16
export type GpuDataId = number ;
17
17
18
- export type GpuArchitecture = 'ampere' ;
18
+ export type GpuArchitecture = 'ampere' | 'gen-12lp' ;
19
19
export type GpuVendor = 'amd' | 'intel' | 'nvidia' ;
20
20
export interface AdapterInfo {
21
21
isArchitecture : ( architecture : GpuArchitecture ) => boolean ;
You can’t perform that action at this time.
0 commit comments