Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -781,7 +781,10 @@ namespace Dml
// this branch could be reached with a bad custom operator or malformed file. If
// a legitimate case reaches here and DML needs to support a new input/output type
// besides tensors, then remove the assert.
assert(false);

// If the model has nodes that use Optional we will arrive here. It's a valid ONNX model but
// TryGetTensorDataType doesn't handle Optional.
// assert(false);
nodeContainsSupportedDataTypes = false;
return;
}
Expand Down
6 changes: 4 additions & 2 deletions onnxruntime/test/contrib_ops/matmul_4bits_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -527,18 +527,20 @@ void RunTest(int64_t M, int64_t N, int64_t K, int64_t block_size, bool has_zerop
if (std::is_same_v<T, MLFloat16>) {
#ifdef USE_CUDA
execution_providers.push_back(DefaultCudaExecutionProvider());
RunTest<MLFloat16>(opts, std::move(execution_providers));
#endif
#ifdef USE_ROCM
execution_providers.push_back(DefaultRocmExecutionProvider());
RunTest<MLFloat16>(opts, std::move(execution_providers));
#endif
#ifdef USE_DML
execution_providers.push_back(DefaultDmlExecutionProvider());
RunTest<MLFloat16>(opts, std::move(execution_providers));
#endif
#ifdef USE_WEBGPU
execution_providers.push_back(DefaultWebGpuExecutionProvider());
#endif

RunTest<MLFloat16>(opts, std::move(execution_providers));
#endif
} else {
#ifdef USE_ROCM
execution_providers.push_back(DefaultRocmExecutionProvider());
Expand Down
3 changes: 2 additions & 1 deletion onnxruntime/test/providers/cpu/math/softmax_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@ TEST(SoftmaxOperator, webgpu_nan) {
test.AddOutput<float>("Y", dimensions, expected_result);

// explicitly disable for EPs that do not handle NaN
test.Run(OpTester::ExpectResult::kExpectSuccess, "", {kCpuExecutionProvider, kCoreMLExecutionProvider});
test.Run(OpTester::ExpectResult::kExpectSuccess, "",
{kCpuExecutionProvider, kCoreMLExecutionProvider, kDmlExecutionProvider});
}
#endif

Expand Down
Loading