← 返回标签列表

#lecture

1 篇文章

cuda_mode_1

hello load inline这个是torch加载C++扩展的简单demo。代码比较简单 123456789101112131415161718import torchfrom torch.utils.cpp_extension import load_inlinecpp_source = &q...