We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 639857f commit 8e85d72Copy full SHA for 8e85d72
paddle/phi/kernels/gpu/lstsq_kernel.cu
@@ -16,6 +16,7 @@
16
#include <algorithm>
17
#include <complex>
18
19
+#include "paddle/phi/backends/gpu/cuda/cudnn_workspace_helper.h"
20
#include "paddle/phi/backends/gpu/gpu_context.h"
21
#include "paddle/phi/core/kernel_registry.h"
22
#include "paddle/phi/kernels/full_kernel.h"
@@ -61,6 +62,10 @@ void LstsqKernel(const Context& dev_ctx,
61
62
singular_values);
63
return;
64
}
65
+
66
+ CUDNN_ENFORCE_TENSOR_SIZE_SUPPORTED(x);
67
+ CUDNN_ENFORCE_TENSOR_SIZE_SUPPORTED(y);
68
69
auto x_dims = x.dims();
70
auto y_dims = y.dims();
71
int dim_size = x_dims.size();
0 commit comments