#cuda

cuda_mode_1

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