2025年1月12日
pytorch算子调用过程：以rand算子为例
摘要： 通过pytorch的torch.profiler带调用栈采集运行过程可以看到如下信息，通过chrome://tracing查看， 图上每个小条条表示一个traceEvent, json中的信息如下图所示，其中cat表示traceEvent的类型，有cpu_op,python_function等，前者 阅读全文
posted @ 2025-01-12 18:32 zwlwf 阅读(151) 评论(0) 推荐(0)