From 3dc638bffb4b0d3a718a86e33b16708532f28bd3 Mon Sep 17 00:00:00 2001 From: dc-dc-dc Date: Tue, 2 Jan 2024 18:50:34 -0500 Subject: [PATCH] remove monostate --- python/src/ops.cpp | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/python/src/ops.cpp b/python/src/ops.cpp index 18151daad..0c91f0c0e 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -3198,12 +3198,9 @@ void init_ops(py::module_& m) { "tensordot", [](const array& a, const array& b, - const std::variant>>& - dims, + const std::variant>>& dims, StreamOrDevice s) { - if (std::holds_alternative(dims)) { - return tensordot(a, b, 2, s); - } else if (auto pv = std::get_if(&dims); pv) { + if (auto pv = std::get_if(&dims); pv) { return tensordot(a, b, *pv, s); } else { return tensordot(