chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,14 @@
|
||||
cmake_minimum_required(VERSION 3.15)
|
||||
project(test_paddle_lib)
|
||||
|
||||
list(APPEND CMAKE_PREFIX_PATH "@PADDLE_BINARY_DIR@/paddle_install_dir")
|
||||
find_package(Paddle REQUIRED)
|
||||
include_directories(${PADDLE_INCLUDE_DIRS})
|
||||
|
||||
add_executable(test_paddle_lib test_paddle_lib.cc)
|
||||
target_link_libraries(test_paddle_lib ${PADDLE_LIBRARIES})
|
||||
|
||||
if(@WITH_GPU@)
|
||||
add_executable(test_paddle_lib_gpu test_paddle_lib_gpu.cc)
|
||||
target_link_libraries(test_paddle_lib_gpu ${PADDLE_LIBRARIES})
|
||||
endif()
|
||||
@@ -0,0 +1,27 @@
|
||||
// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
#include <cassert>
|
||||
|
||||
#include "paddle/extension.h"
|
||||
|
||||
int main() {
|
||||
int data[] = {1, 2, 3, 4};
|
||||
auto tensor = paddle::from_blob(data, {2, 2}, phi::DataType::INT32);
|
||||
|
||||
assert(tensor.numel() == 4);
|
||||
assert(tensor.dtype() == phi::DataType::INT32);
|
||||
assert(tensor.is_cpu());
|
||||
assert(tensor.template data<int>() == data);
|
||||
}
|
||||
@@ -0,0 +1,25 @@
|
||||
// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
#include <cassert>
|
||||
|
||||
#include "paddle/extension.h"
|
||||
|
||||
int main() {
|
||||
float data[] = {1., 2., 3., 4.};
|
||||
auto tensor = paddle::from_blob(data, {2, 2}, phi::DataType::FLOAT32);
|
||||
auto gpu_tensor =
|
||||
paddle::experimental::copy_to(tensor, phi::GPUPlace(), false);
|
||||
assert(gpu_tensor.is_gpu());
|
||||
}
|
||||
Reference in New Issue
Block a user