Protonu comments

Repositories
Issues
Comments

Results 2 comments of


                                            Protonu

Cannot correctly compute cross entropy loss (backward) when the reduction used is "sum" and torch executor is used.

A similar issue that shows up is this code: ``` from collections.abc import Sequence import itertools from functools import partial from typing import Any # NOTE: Dependency on fdm and...

Cannot correctly compute cross entropy loss (backward) when the reduction used is "sum" and torch executor is used.

The interesting thing is it for the nvfuser backend it fails, but if I turn on the custom decomposition for cross entropy loss (for nvFuser) it works correctly.