saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Disable CudaKernelTest.SoftmaxCrossEntropyLoss_LargeSizeTensor because it's flaky.

Browse source
This commit is contained in:
Edward Chen 2020-04-21 20:30:45 +00:00
parent 28a0c863b1
commit 2e4b9b1d0e
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
orttraining/orttraining/test/training_ops/cuda/cross_entropy_test.cc
View file

@ -363,7 +363,9 @@ TEST(CudaKernelTest, SoftmaxCrossEntropyLoss_MediumSizeTensor) {
TestSoftmaxCrossEntropyLoss(&X_dims, &index_dims, nullptr, &Y_dims_none, &log_prob_dims, "none");
}
TEST(CudaKernelTest, SoftmaxCrossEntropyLoss_LargeSizeTensor) {
// TODO fix flaky test (https://msdata.visualstudio.com/Vienna/_workitems/edit/596949)
// failing random seed: 2873512643
TEST(CudaKernelTest, DISABLED_SoftmaxCrossEntropyLoss_LargeSizeTensor) {
std::vector<int64_t> X_dims{4, 512, 30528};
std::vector<int64_t> index_dims{4, 30528};
std::vector<int64_t> weight_dims{512};