03 2023 档案

摘要:算子开发## ONNX导出 ### torch.onnx.export 细解 `torch.onnx.export`中需要的模型实际上是一个`torch.jit.ScriptModule`。而要把普通 PyTorch 模型转一个这样的`TorchScript`模型,有跟踪(trace)和脚本化(script 阅读全文
posted @ 2023-03-29 16:16 nanmi 编辑
摘要:CUDA优化# CUDA优化 ![image](https://img2023.cnblogs.com/blog/1990650/202306/1990650-20230625164824401-216646758.png) ## 避免Warp分化 ```C++ if (con) { //do somethin 阅读全文
posted @ 2023-03-14 17:09 nanmi 阅读(164) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示