mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-17 06:38:38 +08:00
fix typo: accross -> across (#1609)
Signed-off-by: Cocoa <i@uwucocoa.moe>
This commit is contained in:
@@ -495,15 +495,15 @@ TEST_CASE("test compile tape with outside parents") {
|
||||
}
|
||||
}
|
||||
|
||||
auto compile_accross_streams(const std::vector<array>& inputs) {
|
||||
auto compile_across_streams(const std::vector<array>& inputs) {
|
||||
auto s2 = new_stream(default_device());
|
||||
auto x = exp(abs(inputs[0]));
|
||||
auto y = exp(abs(x, s2), s2);
|
||||
return std::vector<array>{y};
|
||||
}
|
||||
|
||||
TEST_CASE("test compile accross streams") {
|
||||
auto cfun = compile(compile_accross_streams);
|
||||
TEST_CASE("test compile across streams") {
|
||||
auto cfun = compile(compile_across_streams);
|
||||
auto x = array({2.0f});
|
||||
auto out = cfun({x})[0];
|
||||
auto& p1 = out.primitive();
|
||||
|
Reference in New Issue
Block a user