diff --git a/docs/src/dev/extensions.rst b/docs/src/dev/extensions.rst index 35f493fac..35c655f6c 100644 --- a/docs/src/dev/extensions.rst +++ b/docs/src/dev/extensions.rst @@ -485,7 +485,7 @@ each data type. instantiate_axpby(float32, float); instantiate_axpby(float16, half); - instantiate_axpby(bflot16, bfloat16_t); + instantiate_axpby(bfloat16, bfloat16_t); instantiate_axpby(complex64, complex64_t); This kernel will be compiled into a metal library ``mlx_ext.metallib`` as we diff --git a/examples/extensions/axpby/axpby.metal b/examples/extensions/axpby/axpby.metal index 64980578f..03b373c99 100644 --- a/examples/extensions/axpby/axpby.metal +++ b/examples/extensions/axpby/axpby.metal @@ -59,5 +59,5 @@ template instantiate_axpby(float32, float); instantiate_axpby(float16, half); -instantiate_axpby(bflot16, bfloat16_t); +instantiate_axpby(bfloat16, bfloat16_t); instantiate_axpby(complex64, complex64_t); \ No newline at end of file