TadaoYamaoka https://blog.hatena.ne.jp/TadaoYamaoka/ TadaoYamaokaの開発日記 https://tadaoyamaoka.hatenablog.com/ dlshogi torch.compile は、PyTorch 2.0 以降で導入された高速化機能で、既存の PyTorch コードをほとんど変更せずに JIT コンパイルして最適化できる仕組みである。 主に GPU 実行時のオーバーヘッド削減やカーネル融合によって性能を向上できる。dlshogiの train.py と ptl.py を torch.compile に対応して、どれくらい学習が速くなるか測定した。 torch.compile対応の実装 「--use_compile」オプションで、torch.compileの有効無効を切り替えられるようにした。Pytorchのtorch.compileの引数に… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Ftadaoyamaoka.hatenablog.com%2Fentry%2F2026%2F05%2F09%2F210912" title="【dlshogi】torch.compileに対応したら学習が1.6倍速くなった件 - TadaoYamaokaの開発日記" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> https://cdn-ak.f.st-hatena.com/images/fotolife/T/TadaoYamaoka/20260509/20260509203341.png Hatena Blog https://hatena.blog 2026-05-09 21:09:12 rich https://tadaoyamaoka.hatenablog.com/entry/2026/05/09/210912 1.0 100%