Use unordered map for kwargs in export/import (#2087)

* use unordered map for kwargs in export/import

* comment
This commit is contained in:
Awni Hannun
2025-04-21 07:17:22 -07:00
committed by GitHub
parent 70ebc3b598
commit dc4eada7f0
4 changed files with 31 additions and 31 deletions

View File

@@ -2,14 +2,14 @@
#pragma once
#include <map>
#include <set>
#include <unordered_map>
#include "mlx/array.h"
namespace mlx::core {
using Args = std::vector<array>;
using Kwargs = std::map<std::string, array>;
using Kwargs = std::unordered_map<std::string, array>;
struct FunctionExporter;