ChatCrawlersearch across public Telegram Open the app
P

Python 中文交流群

сообщение · 2026-01-26 11:00 UTC
A
根本原因在PyTorch.compile的loop是还是跑在Python上面的,只是编译成了C++的C Extension。层次一深cuda->c++->python->c++->cuda的路径上面python段会产生很多开销。 jax.jit编译之后整个函数就都完全不在Python里面跑了,层次深浅对整体影响不大。 (23年的测试了,不知道现在PyTorch有没有优化过了)

Вся лента · оригинал в Telegram

Open in Telegram Каталог площадок Искать в ChatCrawler

A snapshot of an open public feed from the search index ChatCrawler — “Google for public Telegram”; refreshed as the venue is crawled. Times are UTC.

Public content only, official Telegram API. About · FAQ · What we do not do · Remove a page · Catalog