未验证 提交 69a01c47 编写于 作者: J jiangcheng 提交者: GitHub

fix NetBuilder API Name bug in cinn_lib_test (#40392)

* fix NetBuilder API Name bug in cinn_lib_test

* update cinn version to newest
上级 70f83f1d
......@@ -26,7 +26,7 @@ add_definitions(-w)
######################################
include(ExternalProject)
set(CINN_PREFIX_DIR ${THIRD_PARTY_PATH}/CINN)
set(CINN_GIT_TAG release/v0.1)
set(CINN_GIT_TAG 56879b637e2c4db19091eedad03d7cc674e092a2)
set(CINN_OPTIONAL_ARGS -DPY_VERSION=${PY_VERSION}
-DWITH_CUDA=${WITH_GPU}
-DWITH_CUDNN=${WITH_GPU}
......
......@@ -45,8 +45,8 @@ Program CreateAddProgram() {
NetBuilder builder("net_builder");
auto a = builder.CreateInput(Float(32), {M, N});
auto b = builder.CreateInput(Float(32), {M, N});
auto c = builder.add(a, b);
auto d = builder.add(a, c);
auto c = builder.Add(a, b);
auto d = builder.Add(a, c);
auto program = builder.Build();
return program;
......@@ -116,8 +116,8 @@ TEST(net_build, program_execute_fc) {
auto w = builder.CreateInput(Float(32), {N, K}, "W"); // weight
auto b = builder.CreateInput(Float(32), {N}, "B"); // bias
auto mul_out = builder.mul(a, w, 2, 1);
auto add_out = builder.add(mul_out, b);
auto mul_out = builder.Mul(a, w, 2, 1);
auto add_out = builder.Add(mul_out, b);
auto program = builder.Build();
#ifdef PADDLE_WITH_CUDA
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册