03 2023 档案
摘要:
## ONNX导出 ### torch.onnx.export 细解 `torch.onnx.export`中需要的模型实际上是一个`torch.jit.ScriptModule`。而要把普通 PyTorch 模型转一个这样的`TorchScript`模型,有跟踪(trace)和脚本化(script
阅读全文
![算子开发](https://img2023.cnblogs.com/blog/1990650/202303/1990650-20230329161607451-1756677155.png)
摘要:
# CUDA优化 ![image](https://img2023.cnblogs.com/blog/1990650/202306/1990650-20230625164824401-216646758.png) ## 避免Warp分化 ```C++ if (con) { //do somethin
阅读全文
![CUDA优化](https://img2023.cnblogs.com/blog/1990650/202303/1990650-20230314170853591-1356080928.png)