forked from jiuyuan/InfiniTensor
12 lines
322 B
C
12 lines
322 B
C
|
#pragma once
|
||
|
|
|
||
|
|
namespace infini {
|
||
|
|
|
||
|
|
void _sg2bmm(float *__restrict__ q, float *__restrict__ k,
|
||
|
|
float *__restrict__ y, int bs, int n, int m, int w, int d);
|
||
|
|
|
||
|
|
void _sgbmml(float *__restrict__ q, float *__restrict__ k,
|
||
|
|
float *__restrict__ y, int bs, int n, int m, int w, int d);
|
||
|
|
|
||
|
|
} // namespace infini
|