extension-cpp

extension-cpp copied to clipboard

mycode -setup.py -a_cuda_kernel.cu -a_cuda.cpp I need to open rdc mode，so from setuptools import setup from torch.utils.cpp_extension import BuildExtension, CUDAExtension setup( name='a_cuda', ext_modules=[ CUDAExtension('a_cuda', [ 'a_cuda.cpp', 'a_cuda_kernel.cu', ],extra_compile_args={'nvcc':['-rdc','true']}), ], cmdclass={ 'build_ext':...

aisuta

cuda does not install

18

python : 3.7 cuda : 11.1 pytorch : 1.8 I am trying to compile the cuda code which does not work, could you have a look please? thanks ``` Traceback...

dorooddorood606

- OS: win10 - PyTorch version: 1.10 - How you installed PyTorch (conda, pip, source): pip - Python version: 3.9 - CUDA/cuDNN version: 11.3 - GPU models and configuration: 1650...

Arsmart123

Writing a custom C++ extension with variants for both CPU and GPU?

2

Hi. I am interested in writing a custom C++/CUDA extension. The tutorial [here](https://pytorch.org/tutorials/advanced/cpp_extension.html) only shows a scenario where you have a pure CUDA kernel which will not work on a...

MalayAgr

Hi! I am wondering how to debug in such environment? I have tried to insert a "printf("hello wolrd")" sentence in .cu file, but it compiles failure! If I delete it,...

Arsmart123

JIT cpp extension fails to find cl.exe on windows?

5

I am trying to use the JIT cpp extension, and have a script from my friend, who is able to run it in linux, but whenever I run it on...

tueboesen

I think there's a mistake in the cpp section and I want to ask before I "fix" it. It doesn't technically cause an error, but it makes it harder to...

stevenwalton

CUDA version Error

3

PyTorch GitHub Issues Guidelines -------------------------------- We like to limit our issues to bug reports and feature requests. If you have a question or would like help and support, please visit...

hao6699

Compilation error of cuda/setup.py on windows

11

Hi, I tried to build the extension using your sc. I managed to build from cpp/setup.py and was able to import it. However, I just cannot build from cuda/setup.py. I...

Xieyuan0018

ImportError: No module named 'lltm_cuda'

1

Hi,Peter CMD: python3 jit2.py no result,help, pls ... Loading extension module lltm_cuda... Traceback (most recent call last): File "jit2.py", line 24, in verbose=True File "E:\Python36\lib\site-packages\torch\utils\cpp_extension.py", line 645, i n load...

liuchanfeng165