File size: 517 Bytes
712dbf0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
// Copyright © 2023 Apple Inc.
#define DOCTEST_CONFIG_IMPLEMENT
#include "doctest/doctest.h"
#include <cstdlib>
#include "mlx/mlx.h"
using namespace mlx::core;
int main(int argc, char** argv) {
doctest::Context context;
const char* device = std::getenv("DEVICE");
if (device != nullptr && std::string(device) == "cpu") {
set_default_device(Device::cpu);
} else if (metal::is_available()) {
set_default_device(Device::gpu);
}
context.applyCommandLine(argc, argv);
return context.run();
}
|