OpInfo for nn.functional.cross_entropy#63547
OpInfo for nn.functional.cross_entropy#63547krshrimali wants to merge 16 commits intopytorch:masterfrom
nn.functional.cross_entropy#63547Conversation
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 3e59876 (more details on the Dr. CI page): 💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Please report bugs/suggestions to the (internal) Dr. CI Users group. |
|
@zou3519 The failing tests here look awful lot like the ones in #63854. Given that |
CI Flow Status⚛️ CI FlowRuleset - Version:
You can add a comment to the PR and tag @pytorchbot with the following commands: # ciflow rerun, "ciflow/default" will always be added automatically
@pytorchbot ciflow rerun
# ciflow rerun with additional labels "-l <ciflow/label_name>", which is equivalent to adding these labels manually and trigger the rerun
@pytorchbot ciflow rerun -l ciflow/scheduled -l ciflow/slowFor more information, please take a look at the CI Flow Wiki. |
zou3519
left a comment
There was a problem hiding this comment.
This looks pretty good. I had one suggestion for improving the test case, please let me know what you think
| skips=( | ||
| # AssertionError: False is not true : Scalars failed to compare as equal! 0 != 1536 | ||
| # test_ops.TestJitCUDA.test_variant_consistency_jit_nn_functional_cross_entropy_cuda_float32 leaked | ||
| # 1536 bytes CUDA memory on device 0 |
|
@zou3519 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
|
@zou3519 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
Reference: pytorch/functorch#78 and #54261
TODOs:
test_nn.pyforcross_entropy.cc: @mruberry @zou3519
cc @albanD @mruberry @jbschlosser @walterddr