@@ -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