From 8b532b3e30d05a16500bafb8697dd29bfe479a02 Mon Sep 17 00:00:00 2001 From: "Zheng, Zhaoqiong" Date: Mon, 9 Jun 2025 15:25:33 +0800 Subject: [PATCH] fix pybind11 errors in cuda extension example --- extension_cpp/csrc/cuda/muladd.cu | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/extension_cpp/csrc/cuda/muladd.cu b/extension_cpp/csrc/cuda/muladd.cu index 6700d08..570e5d1 100644 --- a/extension_cpp/csrc/cuda/muladd.cu +++ b/extension_cpp/csrc/cuda/muladd.cu @@ -1,4 +1,6 @@ -#include +#include +#include +#include #include #include @@ -18,7 +20,7 @@ at::Tensor mymuladd_cuda(const at::Tensor& a, const at::Tensor& b, double c) { TORCH_INTERNAL_ASSERT(b.device().type() == at::DeviceType::CUDA); at::Tensor a_contig = a.contiguous(); at::Tensor b_contig = b.contiguous(); - at::Tensor result = torch::empty(a_contig.sizes(), a_contig.options()); + at::Tensor result = at::empty(a_contig.sizes(), a_contig.options()); const float* a_ptr = a_contig.data_ptr(); const float* b_ptr = b_contig.data_ptr(); float* result_ptr = result.data_ptr(); @@ -41,7 +43,7 @@ at::Tensor mymul_cuda(const at::Tensor& a, const at::Tensor& b) { TORCH_INTERNAL_ASSERT(b.device().type() == at::DeviceType::CUDA); at::Tensor a_contig = a.contiguous(); at::Tensor b_contig = b.contiguous(); - at::Tensor result = torch::empty(a_contig.sizes(), a_contig.options()); + at::Tensor result = at::empty(a_contig.sizes(), a_contig.options()); const float* a_ptr = a_contig.data_ptr(); const float* b_ptr = b_contig.data_ptr(); float* result_ptr = result.data_ptr();