#include <stdio.h>
#include "mlx/c/mlx.h"
void print_array(const char* msg, mlx_array arr) {
mlx_string str = mlx_string_new();
mlx_array_tostring(&str, arr);
printf("%s\n%s\n", msg, mlx_string_data(str));
mlx_string_free(str);
}
int main() {
mlx_stream stream = mlx_default_cpu_stream_new();
double data[] = {1, 2, 3, 4, 5, 6};
int shape[] = {2, 3};
mlx_array arr = mlx_array_new_data(data, shape, 2, MLX_FLOAT64);
print_array("hello world!", arr);
mlx_array two = mlx_array_new_int(2);
mlx_divide(&arr, arr, two, stream);
print_array("divive by 2!", arr);
mlx_arange(&arr, 0, 3, 0.5, MLX_FLOAT64, stream);
print_array("arange", arr);
mlx_array_free(arr);
mlx_array_free(two);
mlx_stream_free(stream);
return 0;
}