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

Ring distributed backend (#1784)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-01-27 22:15:01 -08:00
committed by GitHub
parent 2235dee906
commit ccb61d7aae
17 changed files with 1078 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mlx/ops.h
View File

@@ -222,7 +222,7 @@ split(const array& a, const Shape& indices, StreamOrDevice s = {});
std::vector<array> meshgrid(
const std::vector<array>& arrays,
bool sparse = false,
std::string indexing = "xy",
const std::string& indexing = "xy",
StreamOrDevice s = {});
/**
@@ -274,7 +274,7 @@ array pad(
const Shape& low_pad_size,
const Shape& high_pad_size,
const array& pad_value = array(0),
const std::string mode = "constant",
const std::string& mode = "constant",
StreamOrDevice s = {});
/** Pad an array with a constant value along all axes */
@@ -282,19 +282,19 @@ array pad(
const array& a,
const std::vector<std::pair<int, int>>& pad_width,
const array& pad_value = array(0),
const std::string mode = "constant",
const std::string& mode = "constant",
StreamOrDevice s = {});
array pad(
const array& a,
const std::pair<int, int>& pad_width,
const array& pad_value = array(0),
const std::string mode = "constant",
const std::string& mode = "constant",
StreamOrDevice s = {});
array pad(
const array& a,
int pad_width,
const array& pad_value = array(0),
const std::string mode = "constant",
const std::string& mode = "constant",
StreamOrDevice s = {});
/** Permutes the dimensions in reverse order. */