zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

use string (#976)

Browse Source
This commit is contained in:
Awni Hannun
2024-04-09 11:22:00 -07:00
committed by GitHub
parent b63ef10a7f
commit ae812350f9
4 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/linalg.cpp
View File

@@ -96,7 +96,7 @@ inline array matrix_norm(
inline array matrix_norm(
const array& a,
std::string_view ord,
const std::string& ord,
const std::vector<int>& axis,
bool keepdims,
StreamOrDevice s) {
@@ -153,7 +153,7 @@ array norm(
array norm(
const array& a,
std::string_view ord,
const std::string& ord,
const std::optional<std::vector<int>>& axis /* = std::nullopt */,
bool keepdims /* = false */,
StreamOrDevice s /* = {} */) {