提交 bda20086 编写于 作者: Y Yu Yang

Add TODO for GPU unittest

上级 87dfc12a
......@@ -29,6 +29,7 @@ static std::vector<paddle_real> randomBuffer(size_t bufSize) {
}
TEST(CAPIArguments, create) {
//! TODO(yuyang18): Test GPU Code.
paddle_arguments args = paddle_arguments_create_none();
uint64_t size;
ASSERT_EQ(kPD_NO_ERROR, paddle_arguments_size(args, &size));
......
......@@ -33,6 +33,7 @@ static std::vector<paddle_real> randomBuffer(size_t bufSize) {
}
TEST(GradientMachine, testPredict) {
//! TODO(yuyang18): Test GPU Code.
paddle::TrainerConfigHelper config("./test_predict_network.py");
std::string buffer;
ASSERT_TRUE(config.getModelConfig().SerializeToString(&buffer));
......
......@@ -16,6 +16,7 @@ limitations under the License. */
#include "gtest/gtest.h"
TEST(CAPIMatrix, create) {
//! TODO(yuyang18): Test GPU Code.
paddle_matrix mat = paddle_matrix_create(128, 32, false);
std::vector<paddle_real> sampleRow;
sampleRow.resize(32);
......
......@@ -16,6 +16,7 @@ limitations under the License. */
#include "gtest/gtest.h"
TEST(CAPIVector, create) {
//! TODO(yuyang18): Test GPU Code.
paddle_ivector vec;
int array[3] = {1, 2, 3};
vec = paddle_ivector_create(array, 3, true, false);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册