← 返回标签列表

#lecture

1 篇文章

cuda_mode_1

hello load inline这个是torch加载C++扩展的简单demo。代码比较简单 import torch from torch.utils.cpp_extension import load_inline cpp_source = """ std::st...