提交 c2147606 编写于 作者: L liuruilong

modify unit testC

上级 69e88c5f
......@@ -1188,6 +1188,11 @@ void Gemm::WriteWithBnAddRelu(int mc, int nc, float *c, float *C, int ldc,
}
}
void Gemm::VectorKernel(int m, int n, int k, float alpha, const float *A,
int lda, const float *B, int ldb, float beta, float *C,
int ldc, bool relu) {
}
#else
void Gemm::AddDot4x4(int k, const float *a, const float *b, float *c, int ldc) {
......
......@@ -23,10 +23,11 @@ int main() {
// ../../../test/models/googlenet
// ../../../test/models/mobilenet
// auto program = loader.Load(g_mobilenet_ssd, true);
std::string g_super = "../models/superresoltion";
// auto program = loader.Load(g_super, true);
// auto program = loader.Load(std::string(g_ocr) + "/model",
// std::string(g_ocr) + "/params", false);
auto program = loader.Load(std::string(g_super) + "/model",
std::string(g_super) + "/params", false);
// program.originProgram->Description("program desc: ");
return 0;
......
......@@ -18,7 +18,7 @@ limitations under the License. */
#include "../test_include.h"
int main() {
paddle_mobile::PaddleMobile<paddle_mobile::GPU_CL> paddle_mobile;
paddle_mobile::PaddleMobile<paddle_mobile::CPU> paddle_mobile;
// paddle_mobile.SetThreadNum(4);
auto time1 = paddle_mobile::time();
#ifdef PADDLE_MOBILE_CL
......@@ -42,7 +42,7 @@ int main() {
std::vector<float> vec_result;
auto time3 = paddle_mobile::time();
int max = 10;
int max = 1;
for (int i = 0; i < max; ++i) {
vec_result = paddle_mobile.Predict(input, dims);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册