mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
typos
This commit is contained in:
@@ -1,6 +1,6 @@
|
|||||||
// Copyright © 2025 Apple Inc.
|
// Copyright © 2025 Apple Inc.
|
||||||
|
|
||||||
// This file must not include any host-only code, utilies that work under both
|
// This file must not include any host-only code, utilities that work under both
|
||||||
// host and device can be put here.
|
// host and device can be put here.
|
||||||
//
|
//
|
||||||
// See more about the requirements at:
|
// See more about the requirements at:
|
||||||
@@ -202,7 +202,7 @@ struct Limits<
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// CUDA 11 does not have host side arithmatic operators for half types.
|
// CUDA 11 does not have host side arithmetic operators for half types.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
struct Limits<
|
struct Limits<
|
||||||
T,
|
T,
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
// Copyright © 2025 Apple Inc.
|
// Copyright © 2025 Apple Inc.
|
||||||
|
|
||||||
// This file includes host-only utilies for writing CUDA kernels, the difference
|
// This file includes host-only utilities for writing CUDA kernels, the
|
||||||
// from backend/cuda/device/utils.cuh is that the latter file only include
|
// difference from backend/cuda/device/utils.cuh is that the latter file only
|
||||||
// device-only code.
|
// include device-only code.
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
// Copyright © 2025 Apple Inc.
|
// Copyright © 2025 Apple Inc.
|
||||||
|
|
||||||
// This file include utilies that are used by C++ code (i.e. .cpp files).
|
// This file include utilities that are used by C++ code (i.e. .cpp files).
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
|||||||
@@ -319,7 +319,7 @@ class TestCompile(mlx_tests.MLXTestCase):
|
|||||||
# Check the state is unchanged
|
# Check the state is unchanged
|
||||||
self.assertEqual(state["y"], 2)
|
self.assertEqual(state["y"], 2)
|
||||||
|
|
||||||
# Check the udpated state is used
|
# Check the updated state is used
|
||||||
state["y"] = mx.array(3)
|
state["y"] = mx.array(3)
|
||||||
out = test_state(mx.array(1))
|
out = test_state(mx.array(1))
|
||||||
self.assertEqual(out.item(), 4)
|
self.assertEqual(out.item(), 4)
|
||||||
|
|||||||
Reference in New Issue
Block a user