Skip to content

Commit 1a8b523

Browse files
committed
Fix builds
1 parent e398270 commit 1a8b523

File tree

1 file changed

+3
-0
lines changed
  • onnxruntime/core/providers/cuda/math

1 file changed

+3
-0
lines changed

onnxruntime/core/providers/cuda/math/einsum.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ Status Einsum::DeviceCompute(OpKernelContext* context, const std::vector<const T
4646
// EinsumComputeProcessor section -
4747
if (inputs[0]->IsDataType<float>()) {
4848
auto einsum_compute_processor = EinsumTypedComputeProcessor<float>::Create(context, allocator, tp,
49+
nullptr, // mlas_backend_config
4950
*einsum_compute_preprocessor,
5051
&einsum_cuda_assets);
5152

@@ -56,6 +57,7 @@ Status Einsum::DeviceCompute(OpKernelContext* context, const std::vector<const T
5657
return einsum_compute_processor->Run();
5758
} else if (inputs[0]->IsDataType<double>()) {
5859
auto einsum_compute_processor = EinsumTypedComputeProcessor<double>::Create(context, allocator, tp,
60+
nullptr, // mlas_backend_config
5961
*einsum_compute_preprocessor,
6062
&einsum_cuda_assets);
6163

@@ -67,6 +69,7 @@ Status Einsum::DeviceCompute(OpKernelContext* context, const std::vector<const T
6769
return einsum_compute_processor->Run();
6870
} else if (inputs[0]->IsDataType<MLFloat16>()) {
6971
auto einsum_compute_processor = EinsumTypedComputeProcessor<MLFloat16>::Create(context, allocator, tp,
72+
nullptr, // mlas_backend_config
7073
*einsum_compute_preprocessor,
7174
&einsum_cuda_assets);
7275

0 commit comments

Comments
 (0)