[Compat] Add missing interfaces for PyTorch compat (PaddlePaddle#75874)

Author: SigureMo

paddle/phi/api/include/compat/ATen/cuda/CUDAContext.h

@@ -12,7 +12,27 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// #The file has been adapted from pytorch project
+// #Licensed under  BSD-style license -
+// https://github.com/pytorch/pytorch/blob/main/LICENSE
+
 #pragma once
 
 #include <ATen/cuda/Exceptions.h>
+#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 #include <c10/cuda/CUDAStream.h>
+#include <cuda_runtime_api.h>
+#include "paddle/phi/backends/gpu/gpu_info.h"
+
+namespace at::cuda {
+cudaDeviceProp* getDeviceProperties(c10::DeviceIndex device) {
+  return const_cast<cudaDeviceProp*>(
+      &phi::backends::gpu::GetDeviceProperties(device));
+}
+
+cudaDeviceProp* getCurrentDeviceProperties() {
+  auto device = phi::backends::gpu::GetCurrentDeviceId();
+  return getDeviceProperties(device);
+}
+}  // namespace at::cuda
+#endif

paddle/phi/api/include/compat/c10/util/Exception.h

@@ -46,6 +46,9 @@ namespace c10 {
     }                                                                     \
   } while (false);
 
+// Check for a given boolean condition.
+#define CHECK(condition) PD_CHECK(condition, "CHECK failed : ", #condition)
+
 // TORCH_CHECK_OP macro definitions
 #define TORCH_CHECK_EQ(val1, val2) TORCH_CHECK_OP(val1, val2, ==)
 #define TORCH_CHECK_NE(val1, val2) TORCH_CHECK_OP(val1, val2, !=)

paddle/phi/api/include/compat/torch/csrc/api/include/torch/nn/functional.h

@@ -0,0 +1,19 @@
+// Copyright (c) 2025 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.
+
+// #The file has been adapted from pytorch project
+// #Licensed under  BSD-style license -
+// https://github.com/pytorch/pytorch/blob/main/LICENSE
+
+#pragma once

paddle/phi/api/include/compat/torch/csrc/api/include/torch/python.h

@@ -0,0 +1,27 @@
+// Copyright (c) 2025 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.
+
+// #The file has been adapted from pytorch project
+// #Licensed under  BSD-style license -
+// https://github.com/pytorch/pytorch/blob/main/LICENSE
+
+#pragma once
+#include <ATen/Device.h>
+#include <c10/util/Exception.h>
+#include <torch/types.h>
+
+#if !defined(PADDLE_ON_INFERENCE) && !defined(PADDLE_NO_PYTHON)
+// Python bindings for the C++ frontend (includes Python.h)
+#include "paddle/utils/pybind.h"
+#endif

paddle/phi/api/include/compat/torch/extension.h

@@ -0,0 +1,22 @@
+// Copyright (c) 2025 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.
+
+// #The file has been adapted from pytorch project
+// #Licensed under  BSD-style license -
+// https://github.com/pytorch/pytorch/blob/main/LICENSE
+
+#pragma once
+
+#include <torch/all.h>
+#include <torch/python.h>

paddle/phi/api/include/compat/utils/macros.h

@@ -16,9 +16,7 @@
 
 namespace compat {
 #ifndef TORCH_EXTENSION_NAME
-#define _EXPAND(x) x
-#define TORCH_EXTENSION_NAME _EXPAND(PADDLE_EXTENSION_NAME)
-#undef _EXPAND
+#define TORCH_EXTENSION_NAME PADDLE_EXTENSION_NAME
 #endif
 #define UNSUPPORTED_FEATURE_IN_PADDLE(feature) \
   std::cerr << "Unsupported feature in Paddle: " << feature << std::endl;