Correct types for vjp + tests (#418)

* correct types for vjp + tests

* fix build + comment
This commit is contained in:
Awni Hannun
2024-01-10 13:32:37 -08:00
committed by GitHub
parent b7f905787e
commit 3b4f066dac
4 changed files with 75 additions and 4 deletions

View File

@@ -1,5 +1,7 @@
// Copyright © 2023 Apple Inc.
#include <cstring>
#include "mlx/ops.h"
#include "mlx/primitives.h"
#include "mlx/utils.h"