mirror of
https://github.com/ml-explore/mlx.git
synced 2025-08-29 06:29:09 +08:00
fix typo: accross -> across (#1609)
Signed-off-by: Cocoa <i@uwucocoa.moe>
This commit is contained in:
parent
d8c824c594
commit
0d5e7716ad
@ -209,7 +209,7 @@ Metal library by run-time compiling kernels the first time they are used in MLX
|
|||||||
on a given machine. Note run-time compilation incurs a cold-start cost which can
|
on a given machine. Note run-time compilation incurs a cold-start cost which can
|
||||||
be anwywhere from a few hundred millisecond to a few seconds depending on the
|
be anwywhere from a few hundred millisecond to a few seconds depending on the
|
||||||
application. Once a kernel is compiled, it will be cached by the system. The
|
application. Once a kernel is compiled, it will be cached by the system. The
|
||||||
Metal kernel cache persists accross reboots.
|
Metal kernel cache persists across reboots.
|
||||||
|
|
||||||
Troubleshooting
|
Troubleshooting
|
||||||
^^^^^^^^^^^^^^^
|
^^^^^^^^^^^^^^^
|
||||||
|
@ -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 s2 = new_stream(default_device());
|
||||||
auto x = exp(abs(inputs[0]));
|
auto x = exp(abs(inputs[0]));
|
||||||
auto y = exp(abs(x, s2), s2);
|
auto y = exp(abs(x, s2), s2);
|
||||||
return std::vector<array>{y};
|
return std::vector<array>{y};
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("test compile accross streams") {
|
TEST_CASE("test compile across streams") {
|
||||||
auto cfun = compile(compile_accross_streams);
|
auto cfun = compile(compile_across_streams);
|
||||||
auto x = array({2.0f});
|
auto x = array({2.0f});
|
||||||
auto out = cfun({x})[0];
|
auto out = cfun({x})[0];
|
||||||
auto& p1 = out.primitive();
|
auto& p1 = out.primitive();
|
||||||
|
Loading…
Reference in New Issue
Block a user