mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-22 02:58:16 +08:00
Fp64 on the CPU (#1843)
* add fp64 data type * clean build * update docs * fix bug
This commit is contained in:
@@ -151,8 +151,8 @@ void Arange::eval_gpu(const std::vector<array>& inputs, array& out) {
|
||||
case bfloat16:
|
||||
arange_set_scalars<bfloat16_t>(start_, start_ + step_, compute_encoder);
|
||||
break;
|
||||
case complex64:
|
||||
throw std::runtime_error("[Arange::eval_gpu] Does not support complex64");
|
||||
default:
|
||||
throw std::runtime_error("[Arange::eval_gpu] Does not support type.");
|
||||
}
|
||||
|
||||
compute_encoder.set_output_array(out, 2);
|
||||
|
@@ -42,6 +42,9 @@ std::string type_to_name(const Dtype& t) {
|
||||
case float32:
|
||||
tname = "float32";
|
||||
break;
|
||||
case float64:
|
||||
tname = "double";
|
||||
break;
|
||||
case bfloat16:
|
||||
tname = "bfloat16";
|
||||
break;
|
||||
|
Reference in New Issue
Block a user