mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-21 16:51:15 +08:00
Fix typo and missing header (#1266)
This commit is contained in:
parent
8c01a7893b
commit
987785d8d7
@ -124,7 +124,7 @@ std::vector<array> Primitive::vjp(
|
|||||||
const std::vector<int>&,
|
const std::vector<int>&,
|
||||||
const std::vector<array>&) {
|
const std::vector<array>&) {
|
||||||
std::ostringstream msg;
|
std::ostringstream msg;
|
||||||
msg << "[Primitive::vip] Not implemented for ";
|
msg << "[Primitive::vjp] Not implemented for ";
|
||||||
print(msg);
|
print(msg);
|
||||||
msg << ".";
|
msg << ".";
|
||||||
throw std::invalid_argument(msg.str());
|
throw std::invalid_argument(msg.str());
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
|
|
||||||
#include "mlx/array.h"
|
#include "mlx/array.h"
|
||||||
#include "mlx/stream.h"
|
#include "mlx/stream.h"
|
||||||
|
#include "mlx/utils.h"
|
||||||
|
|
||||||
namespace mlx::core::random {
|
namespace mlx::core::random {
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user