forked from jiuyuan/InfiniTensor
17 lines
723 B
C++
17 lines
723 B
C++
#pragma once
|
|
namespace infini {
|
|
namespace opTimer {
|
|
double getPerfConvCudnn(int n, int c, int h, int w, int f, int r, int s,
|
|
int padh, int padw, int strideh, int stridew,
|
|
int dilationh, int dilationw, int group,
|
|
const char *name);
|
|
|
|
double getPerfConvTransposed2dCudnn(int n, int c, int h, int w, int f, int r,
|
|
int s, int padh, int padw, int strideh,
|
|
int stridew, int dilationh, int dilationw,
|
|
int oph, int opw, int group);
|
|
|
|
double getPerfMatmulCublas(int b, int m, int n, int k, const char *name);
|
|
} // namespace opTimer
|
|
} // namespace infini
|