forked from jiuyuan/InfiniTensor
11 lines
217 B
C
11 lines
217 B
C
|
#pragma once
|
||
|
|
|
||
|
|
#include "operators/rms_norm.h"
|
||
|
|
|
||
|
|
namespace infini {
|
||
|
|
|
||
|
|
void rmsnorm_kernel(int dType, void *input, void *weight, void *output,
|
||
|
|
int num_tokens, int hidden_size);
|
||
|
|
|
||
|
|
}; // namespace infini
|