lxchen/Xplace_for_ICCAD
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix compilation bug

Browse Source
This commit is contained in:
liulixinkerry 2022-10-23 18:33:44 +08:00
parent 386bfe5566
commit 00ea219123
3 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cpp_to_py/density_map_cuda/density_map_cuda_kernel.cu
View File

@ -2,6 +2,7 @@
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <torch/extension.h> #include <torch/extension.h>
#include <ATen/cuda/CUDAContext.h>
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <vector> #include <vector>

1
cpp_to_py/density_map_cuda/density_map_cuda_naive_kernel.cu
View File

@ -2,6 +2,7 @@
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <torch/extension.h> #include <torch/extension.h>
#include <ATen/cuda/CUDAContext.h>
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <vector> #include <vector>

1
cpp_to_py/wa_wirelength_hpwl_cuda/wa_wirelength_hpwl_cuda_kernel.cu
View File

@ -2,6 +2,7 @@
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <torch/extension.h> #include <torch/extension.h>
#include <ATen/cuda/CUDAContext.h>
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <vector> #include <vector>