遇到的问题是,代码运行之后一点反应都没有,没有报错,也没有结果。所以就完全不知道这个问题是叫啥问题,无从百度。使用pycharm一步一步debug,发现代码是卡在了fusedact.py中的fused的load这一部分,而这个load从开头的import可以看到,有时从cpp_extension这里导进来的东西。(其实做到现在遇到的问题很多都是c++编译的...
torch.utils.cpp_extension.load(name, sources, extra_cflags=None, extra_cuda_cflags=None, extra_ldflags=None, extra_include_paths=None, build_directory=None, verbose=False) 即时加载(JIT)PyTorch C ++扩展。 为了加载扩展,会创建一个Ninja构建文件,该文件用于将指定的...
JIT编译机制通过调用PyTorch API中的一个简单函数torch.utils.cpp_extension.load(),为你提供了一种即时编译和加载扩展的方式。 fromtorch.utils.cpp_extensionimportloadcppcuda_tutorial=load(name="cppcuda_tutorial",sources=['interpolation.cpp']) 在这里,实际提供的是域setuptools相同的信息。在后台,这将执行...