Skip to content

Commit 57f383c

Browse files
committed
fix ci
1 parent 37139cf commit 57f383c

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

paddle/operators/math/im2col.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ template <class T>
160160
class Col2ImFunctor<paddle::operators::math::ColFormat::kCFO,
161161
platform::CUDADeviceContext, T> {
162162
public:
163-
void operator()(const platform::DeviceContext& context,
163+
void operator()(const platform::CUDADeviceContext& context,
164164
const framework::Tensor& col,
165165
const std::vector<int>& dilation,
166166
const std::vector<int>& stride,

paddle/operators/math/sequence2batch.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ __global__ void CopyMatrixRowsKernel(const T* src, T* dst, const size_t* index,
4141
template <typename T>
4242
class CopyMatrixRowsFunctor<platform::CUDADeviceContext, T> {
4343
public:
44-
void operator()(const platform::CPUDeviceContext& context,
44+
void operator()(const platform::CUDADeviceContext& context,
4545
const framework::Tensor& src, const size_t* index,
4646
framework::Tensor& dst, bool is_src_index) {
4747
auto src_dims = src.dims();

paddle/platform/device_context_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ TEST(Device, Init) {
2424
for (int i = 0; i < count; i++) {
2525
DeviceContext* device_context = new CUDADeviceContext(GPUPlace(i));
2626
Eigen::GpuDevice* gpu_device =
27-
device_context->template GetEigenDevice<GPUPlace>();
27+
device_context->template device_context<CUDADeviceContext>();
2828
ASSERT_NE(nullptr, gpu_device);
2929
delete device_context;
3030
}

0 commit comments

Comments
 (0)