c26ad78f86
### What this PR does / why we need it?
After removing codepsell a while, we discovered that typo had a problem
correctly recognizing certain misspelled words, so I suggested adding it
back.
- vLLM version: v0.14.1
- vLLM main:
d68209402d
---------
Signed-off-by: wangli <wangli858794774@gmail.com>
51 lines
1.8 KiB
C
51 lines
1.8 KiB
C
#ifndef ACLNN_DISPATCH_LAYOUT_H_
|
|
#define ACLNN_DISPATCH_LAYOUT_H_
|
|
|
|
#include "aclnn/acl_meta.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/* function: aclnnDispatchLayoutGetWorkspaceSize
|
|
* topkIdx : required
|
|
* numTokens : required
|
|
* numRanks : required
|
|
* numExperts : required
|
|
* numTopk : required
|
|
* numTokensPerRank : required
|
|
* numTokensPerExpert : required
|
|
* isTokenInRank : required
|
|
* workspaceSize : size of workspace(output).
|
|
* executor : executor context(output).
|
|
*/
|
|
__attribute__((visibility("default"))) aclnnStatus aclnnDispatchLayoutGetWorkspaceSize(
|
|
const aclTensor *topkIdx,
|
|
int64_t numTokens,
|
|
int64_t numRanks,
|
|
int64_t numExperts,
|
|
int64_t numTopk,
|
|
const aclTensor *numTokensPerRank,
|
|
const aclTensor *numTokensPerExpert,
|
|
const aclTensor *isTokenInRank,
|
|
uint64_t *workspaceSize,
|
|
aclOpExecutor **executor);
|
|
|
|
/* function: aclnnDispatchLayout
|
|
* workspace : workspace memory addr(input).
|
|
* workspaceSize : size of workspace(input).
|
|
* executor : executor context(input).
|
|
* stream : acl stream.
|
|
*/
|
|
__attribute__((visibility("default"))) aclnnStatus aclnnDispatchLayout(
|
|
void *workspace,
|
|
uint64_t workspaceSize,
|
|
aclOpExecutor *executor,
|
|
aclrtStream stream);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|