diff --git a/paddle/phi/backends/custom/custom_device_test.cc b/paddle/phi/backends/custom/custom_device_test.cc index 4e786275a631282c2cd7bcdb5f960b0797f4599c..84eaea3426f598de22369ca740d17b77bac38309 100644 --- a/paddle/phi/backends/custom/custom_device_test.cc +++ b/paddle/phi/backends/custom/custom_device_test.cc @@ -18,8 +18,6 @@ #include "paddle/fluid/framework/tensor.h" #include "paddle/fluid/framework/tensor_util.h" -#include "paddle/fluid/framework/variable.h" -#include "paddle/fluid/imperative/gradient_accumulator.h" #include "paddle/fluid/platform/device_context.h" #include "paddle/phi/backends/custom/fake_cpu_device.h" #include "paddle/phi/backends/device_manager.h" @@ -240,51 +238,6 @@ void TestCustomCCL(const paddle::platform::Place& place) { stream); } -void TestBlasAPI(const paddle::platform::Place& place) { - std::cout << "TestBlasAPI on " << place << std::endl; - if (paddle::platform::is_custom_place(place) == false) { - return; - } - auto device = phi::DeviceManager::GetDeviceWithPlace(place); - phi::stream::Stream stream(place, nullptr); - device->BlasAXPBY(stream, 0, 1., nullptr, 1., nullptr); - - paddle::framework::Variable var1; - paddle::framework::Variable var2; - std::vector src_data(10, 1.0); - std::vector dst_data(10, 0.0); - std::vector result; - paddle::platform::CPUPlace src_place; - for (unsigned int i = 0; i < 10; i++) { - result.emplace_back(src_data[i] + dst_data[i]); - } - - std::vector dims = {2, 5}; - auto* src = var1.GetMutable(); - auto* dst = var2.GetMutable(); - src->Resize(phi::make_ddim(dims)); - dst->Resize(phi::make_ddim(dims)); - auto* src_mutable = src->mutable_data(place); - auto* dst_mutable = dst->mutable_data(place); - - paddle::memory::Copy(place, - src_mutable, - src_place, - src_data.data(), - sizeof(float) * src_data.size()); - - paddle::memory::Copy(place, - dst_mutable, - src_place, - dst_data.data(), - sizeof(float) * dst_data.size()); - - paddle::imperative::TensorAdd(var1, &var2); - phi::DenseTensor rlt; - paddle::platform::CPUPlace rlt_place; - paddle::framework::TensorCopySync(*dst, rlt_place, &rlt); -} - TEST(CustomDevice, Tensor) { InitDevice(); auto dev_types = phi::DeviceManager::GetAllDeviceTypes(); @@ -299,7 +252,6 @@ TEST(CustomDevice, Tensor) { TestTensorShareDataWith(place); TestTensorUtils(place); TestCustomCCL(place); - TestBlasAPI(place); } }