diff --git a/docs/build/html/.buildinfo b/docs/build/html/.buildinfo index 630b714d0..5968ca412 100644 --- a/docs/build/html/.buildinfo +++ b/docs/build/html/.buildinfo @@ -1,4 +1,4 @@ # Sphinx build info version 1 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done. -config: 5d79ff0eb4fd2457710e4f29cb9324ed +config: 19a532605596267974106a744b1b96b3 tags: 645f666f9bcd5a90fca523b33c5a78b7 diff --git a/docs/build/html/_sources/install.rst b/docs/build/html/_sources/install.rst index edc3d6143..225ecfd62 100644 --- a/docs/build/html/_sources/install.rst +++ b/docs/build/html/_sources/install.rst @@ -14,7 +14,7 @@ silicon computer is To install from PyPI you must meet the following requirements: - Using an M series chip (Apple silicon) -- Using a native Python >= 3.8 +- Using a native Python >= 3.9 - macOS >= 13.5 .. note:: diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.imag.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.imag.rst new file mode 100644 index 000000000..a443358be --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.imag.rst @@ -0,0 +1,6 @@ +mlx.core.imag +============= + +.. currentmodule:: mlx.core + +.. autofunction:: imag \ No newline at end of file diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.real.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.real.rst new file mode 100644 index 000000000..9128bb410 --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.real.rst @@ -0,0 +1,6 @@ +mlx.core.real +============= + +.. currentmodule:: mlx.core + +.. autofunction:: real \ No newline at end of file diff --git a/docs/build/html/_sources/python/ops.rst b/docs/build/html/_sources/python/ops.rst index e3c50e2ff..572b02a98 100644 --- a/docs/build/html/_sources/python/ops.rst +++ b/docs/build/html/_sources/python/ops.rst @@ -80,6 +80,7 @@ Operations greater_equal hadamard_transform identity + imag inner isfinite isclose @@ -125,6 +126,7 @@ Operations quantize quantized_matmul radians + real reciprocal remainder repeat diff --git a/docs/build/html/_static/documentation_options.js b/docs/build/html/_static/documentation_options.js index 6fe754faf..e8b958b36 100644 --- a/docs/build/html/_static/documentation_options.js +++ b/docs/build/html/_static/documentation_options.js @@ -1,5 +1,5 @@ const DOCUMENTATION_OPTIONS = { - VERSION: '0.18.1', + VERSION: '0.19.0', LANGUAGE: 'en', COLLAPSE_INDEX: false, BUILDER: 'html', diff --git a/docs/build/html/annotated.html b/docs/build/html/annotated.html index 2945f0a6e..6b4b5129b 100644 --- a/docs/build/html/annotated.html +++ b/docs/build/html/annotated.html @@ -123,41 +123,43 @@ $(function(){ initResizable(false); });  CFloor  CGreater  CGreaterEqual - CIntOrFloat - CInTracing - CLeftShift - CLess - CLessEqual - CLog - CLog10 - CLog1p - CLog2 - CLogAddExp - CLogicalAnd - CLogicalNot - CLogicalOr - CMaximum - CMinimum - CMultiply - CNaNEqual - CNegative - CNotEqual - CPower - CRemainder - CRetainGraph - CRightShift - CRound - CRsqrt - CSelect - CSigmoid - CSign - CSin - CSinh - CSqrt - CSquare - CSubtract - CTan - CTanh + CImag + CIntOrFloat + CInTracing + CLeftShift + CLess + CLessEqual + CLog + CLog10 + CLog1p + CLog2 + CLogAddExp + CLogicalAnd + CLogicalNot + CLogicalOr + CMaximum + CMinimum + CMultiply + CNaNEqual + CNegative + CNotEqual + CPower + CReal + CRemainder + CRetainGraph + CRightShift + CRound + CRsqrt + CSelect + CSigmoid + CSign + CSin + CSinh + CSqrt + CSquare + CSubtract + CTan + CTanh  ►Ndistributed  CAllGather  CAllReduce @@ -251,63 +253,65 @@ $(function(){ initResizable(false); });  CGreater  CGreaterEqual  CHadamard - CInverse - CLess - CLessEqual - CLoad - CLog - CLog1p - CLogAddExp - CLogicalAnd - CLogicalNot - CLogicalOr - CMatmul - CMaximum - CMinimum - CMultiply - CNegative - CNodeNamer - CNotEqual - CNumberOfElements - CPad - CPartition - CPower - CPrimitive - CPrintFormatter - CQRF - CQuantizedMatmul - CRandomBits - CReduce - CReductionPlan - CRemainder - CReshape - CRound - CScan - CScatter - CSelect - CSigmoid - CSign - CSin - CSinh - CSlice - CSliceUpdate - CSoftmax - CSort - CSplit - CSqrt - CSquare - CStopGradient - CStream - CStreamContext - CSubtract - CSVD - CTan - CTanh - CTranspose - CTypeToDtype - CUnaryPrimitive - CUniform - CView + CImag + CInverse + CLess + CLessEqual + CLoad + CLog + CLog1p + CLogAddExp + CLogicalAnd + CLogicalNot + CLogicalOr + CMatmul + CMaximum + CMinimum + CMultiply + CNegative + CNodeNamer + CNotEqual + CNumberOfElements + CPad + CPartition + CPower + CPrimitive + CPrintFormatter + CQRF + CQuantizedMatmul + CRandomBits + CReal + CReduce + CReductionPlan + CRemainder + CReshape + CRound + CScan + CScatter + CSelect + CSigmoid + CSign + CSin + CSinh + CSlice + CSliceUpdate + CSoftmax + CSort + CSplit + CSqrt + CSquare + CStopGradient + CStream + CStreamContext + CSubtract + CSVD + CTan + CTanh + CTranspose + CTypeToDtype + CUnaryPrimitive + CUniform + CView  ►Nsteel  CAccumHelper  ►CBlockLoader @@ -412,75 +416,77 @@ $(function(){ initResizable(false); });  CGEMVTKernelVector matrix multiplication  CGreater  CGreaterEqual - CIndices - CKernelMergeSort - CKernelMultiBlockMergeSort - CLeftShift - CLess - CLessEqual - CLessThan - CLimits - CLimits< bfloat16_t > - CLimits< bool > - CLimits< complex64_t > - CLimits< float > - CLimits< half > - CLimits< int16_t > - CLimits< int32_t > - CLimits< int64_t > - CLimits< int8_t > - CLimits< uint16_t > - CLimits< uint32_t > - CLimits< uint64_t > - CLimits< uint8_t > - CLog - CLog10 - CLog1p - CLog2 - CLogAddExp - CLogicalAnd - CLogicalNot - CLogicalOr - Clooped_elem_to_loc - Clooped_elem_to_loc< 0, offset_t > - Clooped_elem_to_loc< 1, offset_t > - CMax - CMaximum - CMin - CMinimum - Cmlx_atomic - Cmlx_atomic< T, enable_if_t< is_metal_atomic< T > > > - CMLXConvParams - CMLXFastAttentionParams - CMLXScaledDotProductAttentionParams - CMultiply - CNaNEqual - CNegative - CNone - CNotEqual - COr - CPower - CProd - CQuantizedBlockLoader - CReadWriter - CRemainder - CRightShift - CRound - CRsqrt - CScaleOp - CSelect - CSigmoid - CSign - CSin - CSinh - CSqrt - CSquare - CSubtract - CSum - CTan - CTanh - CThreadPool - CThreadSort + CImag + CIndices + CKernelMergeSort + CKernelMultiBlockMergeSort + CLeftShift + CLess + CLessEqual + CLessThan + CLimits + CLimits< bfloat16_t > + CLimits< bool > + CLimits< complex64_t > + CLimits< float > + CLimits< half > + CLimits< int16_t > + CLimits< int32_t > + CLimits< int64_t > + CLimits< int8_t > + CLimits< uint16_t > + CLimits< uint32_t > + CLimits< uint64_t > + CLimits< uint8_t > + CLog + CLog10 + CLog1p + CLog2 + CLogAddExp + CLogicalAnd + CLogicalNot + CLogicalOr + Clooped_elem_to_loc + Clooped_elem_to_loc< 0, offset_t > + Clooped_elem_to_loc< 1, offset_t > + CMax + CMaximum + CMin + CMinimum + Cmlx_atomic + Cmlx_atomic< T, enable_if_t< is_metal_atomic< T > > > + CMLXConvParams + CMLXFastAttentionParams + CMLXScaledDotProductAttentionParams + CMultiply + CNaNEqual + CNegative + CNone + CNotEqual + COr + CPower + CProd + CQuantizedBlockLoader + CReadWriter + CReal + CRemainder + CRightShift + CRound + CRsqrt + CScaleOp + CSelect + CSigmoid + CSign + CSin + CSinh + CSqrt + CSquare + CSubtract + CSum + CTan + CTanh + CThreadPool + CThreadSort diff --git a/docs/build/html/backend_2common_2ops_8h.html b/docs/build/html/backend_2common_2ops_8h.html index c9928625b..fdaf270c9 100644 --- a/docs/build/html/backend_2common_2ops_8h.html +++ b/docs/build/html/backend_2common_2ops_8h.html @@ -139,6 +139,8 @@ Classes   struct  mlx::core::detail::Floor   +struct  mlx::core::detail::Imag +  struct  mlx::core::detail::Log   struct  mlx::core::detail::Log2 @@ -151,6 +153,8 @@ Classes   struct  mlx::core::detail::Negative   +struct  mlx::core::detail::Real +  struct  mlx::core::detail::Round   struct  mlx::core::detail::Sigmoid diff --git a/docs/build/html/backend_2common_2ops_8h_source.html b/docs/build/html/backend_2common_2ops_8h_source.html index 2dbcfca51..9eeec1ffc 100644 --- a/docs/build/html/backend_2common_2ops_8h_source.html +++ b/docs/build/html/backend_2common_2ops_8h_source.html @@ -514,553 +514,575 @@ $(function(){ initResizable(false); });
297
-
298struct Log {
+
298struct Imag {
299 template <typename T>
-
300 T operator()(T x) {
-
301 return std::log(x);
+
300 T operator()(T x) {
+
301 return std::imag(x);
302 }
303};
304
-
305struct Log2 {
+
305struct Log {
306 template <typename T>
-
307 T operator()(T x) {
-
308 return std::log2(x);
+
307 T operator()(T x) {
+
308 return std::log(x);
309 }
310};
311
-
312struct Log10 {
+
312struct Log2 {
313 template <typename T>
-
314 T operator()(T x) {
-
315 return std::log10(x);
+
314 T operator()(T x) {
+
315 return std::log2(x);
316 }
317};
318
-
319struct Log1p {
+
319struct Log10 {
320 template <typename T>
-
321 T operator()(T x) {
-
322 return log1p(x);
+
321 T operator()(T x) {
+
322 return std::log10(x);
323 }
324};
325
-
326struct LogicalNot {
+
326struct Log1p {
327 template <typename T>
-
328 T operator()(T x) {
-
329 return !x;
+
328 T operator()(T x) {
+
329 return log1p(x);
330 }
331};
332
-
333struct Negative {
+
333struct LogicalNot {
334 template <typename T>
-
335 T operator()(T x) {
-
336 return -x;
+
335 T operator()(T x) {
+
336 return !x;
337 }
338};
339
-
340struct Round {
+
340struct Negative {
341 template <typename T>
-
342 T operator()(T x) {
-
343 return std::rint(x);
+
342 T operator()(T x) {
+
343 return -x;
344 }
-
345
-
-
346 complex64_t operator()(complex64_t x) {
-
347 return {std::rint(x.real()), std::rint(x.imag())};
-
348 }
+
345};
-
349};
+
346
+
+
347struct Real {
+
348 template <typename T>
+
+
349 T operator()(T x) {
+
350 return std::real(x);
+
351 }
-
350
-
-
351struct Sigmoid {
-
352 template <typename T>
-
-
353 T operator()(T x) {
-
354 auto one = static_cast<decltype(x)>(1.0);
-
355 return one / (one + fast_exp(-x));
-
356 }
+
352};
-
357};
+
353
+
+
354struct Round {
+
355 template <typename T>
+
+
356 T operator()(T x) {
+
357 return std::rint(x);
+
358 }
-
358
-
-
359struct Sign {
-
360 template <typename T>
-
-
361 T operator()(T x) {
-
362 return (x > T(0)) - (x < T(0));
-
363 }
+
359
+
+
360 complex64_t operator()(complex64_t x) {
+
361 return {std::rint(x.real()), std::rint(x.imag())};
+
362 }
-
-
364 uint8_t operator()(uint8_t x) {
-
365 return x != 0;
-
366 }
+
363};
+
364
+
+
365struct Sigmoid {
+
366 template <typename T>
-
367 uint16_t operator()(uint16_t x) {
-
368 return x != 0;
-
369 }
+
367 T operator()(T x) {
+
368 auto one = static_cast<decltype(x)>(1.0);
+
369 return one / (one + fast_exp(-x));
+
370 }
-
-
370 uint32_t operator()(uint32_t x) {
-
371 return x != 0;
-
372 }
+
371};
-
-
373 uint64_t operator()(uint64_t x) {
-
374 return x != 0;
-
375 }
+
372
+
+
373struct Sign {
+
374 template <typename T>
+
+
375 T operator()(T x) {
+
376 return (x > T(0)) - (x < T(0));
+
377 }
-
376
-
-
377 complex64_t operator()(complex64_t x) {
-
378 return x == complex64_t(0) ? x : x / std::abs(x);
-
379 }
+
+
378 uint8_t operator()(uint8_t x) {
+
379 return x != 0;
+
380 }
-
380};
+
+
381 uint16_t operator()(uint16_t x) {
+
382 return x != 0;
+
383 }
-
381
-
-
382struct Sin {
-
383 template <typename T>
-
384 T operator()(T x) {
-
385 return std::sin(x);
+
384 uint32_t operator()(uint32_t x) {
+
385 return x != 0;
386 }
-
387};
+
+
387 uint64_t operator()(uint64_t x) {
+
388 return x != 0;
+
389 }
-
388
-
-
389struct Sinh {
-
390 template <typename T>
+
390
-
391 T operator()(T x) {
-
392 return std::sinh(x);
+
391 complex64_t operator()(complex64_t x) {
+
392 return x == complex64_t(0) ? x : x / std::abs(x);
393 }
394};
395
-
396struct Square {
+
396struct Sin {
397 template <typename T>
-
398 T operator()(T x) {
-
399 return x * x;
+
398 T operator()(T x) {
+
399 return std::sin(x);
400 }
401};
402
-
403struct Sqrt {
+
403struct Sinh {
404 template <typename T>
-
405 T operator()(T x) {
-
406 return std::sqrt(x);
+
405 T operator()(T x) {
+
406 return std::sinh(x);
407 }
408};
409
-
410struct Rsqrt {
+
410struct Square {
411 template <typename T>
-
412 T operator()(T x) {
-
413 return static_cast<decltype(x)>(1.0) / std::sqrt(x);
+
412 T operator()(T x) {
+
413 return x * x;
414 }
415};
416
-
417struct Tan {
+
417struct Sqrt {
418 template <typename T>
-
419 T operator()(T x) {
-
420 return std::tan(x);
+
419 T operator()(T x) {
+
420 return std::sqrt(x);
421 }
422};
423
-
424struct Tanh {
+
424struct Rsqrt {
425 template <typename T>
-
426 T operator()(T x) {
-
427 return std::tanh(x);
+
426 T operator()(T x) {
+
427 return static_cast<decltype(x)>(1.0) / std::sqrt(x);
428 }
429};
430
-
431struct Add {
+
431struct Tan {
432 template <typename T>
-
433 T operator()(T x, T y) {
-
434 return x + y;
+
433 T operator()(T x) {
+
434 return std::tan(x);
435 }
436};
437
-
438struct Divide {
+
438struct Tanh {
439 template <typename T>
-
440 T operator()(T x, T y) {
-
441 return x / y;
+
440 T operator()(T x) {
+
441 return std::tanh(x);
442 }
443};
444
-
445struct Remainder {
+
445struct Add {
446 template <typename T>
-
447 std::enable_if_t<std::is_integral_v<T> & !std::is_signed_v<T>, T> operator()(
-
448 T numerator,
-
449 T denominator) {
-
450 return numerator % denominator;
-
451 }
+
447 T operator()(T x, T y) {
+
448 return x + y;
+
449 }
-
452
+
450};
+
+
451
+
+
452struct Divide {
453 template <typename T>
-
454 std::enable_if_t<std::is_integral_v<T> & std::is_signed_v<T>, T> operator()(
-
455 T numerator,
-
456 T denominator) {
-
457 auto r = numerator % denominator;
-
458 if (r != 0 && (r < 0 != denominator < 0))
-
459 r += denominator;
-
460 return r;
-
461 }
+
454 T operator()(T x, T y) {
+
455 return x / y;
+
456 }
-
462
-
463 template <typename T>
-
-
464 std::enable_if_t<!std::is_integral_v<T>, T> operator()(
-
465 T numerator,
-
466 T denominator) {
-
467 auto r = std::fmod(numerator, denominator);
-
468 if (r != 0 && (r < 0 != denominator < 0)) {
-
469 r += denominator;
-
470 }
-
471 return r;
-
472 }
+
457};
-
473
-
-
474 complex64_t operator()(complex64_t numerator, complex64_t denominator) {
-
475 return numerator % denominator;
-
476 }
+
458
+
+
459struct Remainder {
+
460 template <typename T>
+
+
461 std::enable_if_t<std::is_integral_v<T> & !std::is_signed_v<T>, T> operator()(
+
462 T numerator,
+
463 T denominator) {
+
464 return numerator % denominator;
+
465 }
-
477};
+
466
+
467 template <typename T>
+
+
468 std::enable_if_t<std::is_integral_v<T> & std::is_signed_v<T>, T> operator()(
+
469 T numerator,
+
470 T denominator) {
+
471 auto r = numerator % denominator;
+
472 if (r != 0 && (r < 0 != denominator < 0))
+
473 r += denominator;
+
474 return r;
+
475 }
-
478
-
-
479struct Equal {
-
480 template <typename T>
-
-
481 bool operator()(T x, T y) {
-
482 return x == y;
-
483 }
+
476
+
477 template <typename T>
+
+
478 std::enable_if_t<!std::is_integral_v<T>, T> operator()(
+
479 T numerator,
+
480 T denominator) {
+
481 auto r = std::fmod(numerator, denominator);
+
482 if (r != 0 && (r < 0 != denominator < 0)) {
+
483 r += denominator;
+
484 }
+
485 return r;
+
486 }
-
484};
-
-
485
-
-
486struct NaNEqual {
-
487 template <typename T>
+
487
-
488 bool operator()(T x, T y) {
-
489 return x == y || (std::isnan(x) && std::isnan(y));
+
488 complex64_t operator()(complex64_t numerator, complex64_t denominator) {
+
489 return numerator % denominator;
490 }
491};
492
-
493struct Greater {
+
493struct Equal {
494 template <typename T>
-
495 bool operator()(T x, T y) {
-
496 return x > y;
+
495 bool operator()(T x, T y) {
+
496 return x == y;
497 }
498};
499
-
500struct GreaterEqual {
+
500struct NaNEqual {
501 template <typename T>
-
502 bool operator()(T x, T y) {
-
503 return x >= y;
+
502 bool operator()(T x, T y) {
+
503 return x == y || (std::isnan(x) && std::isnan(y));
504 }
505};
506
-
507struct Less {
+
507struct Greater {
508 template <typename T>
-
509 bool operator()(T x, T y) {
-
510 return x < y;
+
509 bool operator()(T x, T y) {
+
510 return x > y;
511 }
512};
513
-
514struct LessEqual {
+
514struct GreaterEqual {
515 template <typename T>
-
516 bool operator()(T x, T y) {
-
517 return x <= y;
+
516 bool operator()(T x, T y) {
+
517 return x >= y;
518 }
519};
520
-
521struct Maximum {
+
521struct Less {
522 template <typename T>
-
523 std::enable_if_t<std::is_integral_v<T>, T> operator()(T x, T y) {
-
524 return (x > y) ? x : y;
+
523 bool operator()(T x, T y) {
+
524 return x < y;
525 }
-
526
-
527 template <typename T>
-
-
528 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T x, T y) {
-
529 if (std::isnan(x)) {
-
530 return x;
-
531 }
-
532 return (x > y) ? x : y;
-
533 }
+
526};
-
534};
+
527
+
+
528struct LessEqual {
+
529 template <typename T>
+
+
530 bool operator()(T x, T y) {
+
531 return x <= y;
+
532 }
-
535
-
-
536struct Minimum {
-
537 template <typename T>
-
-
538 std::enable_if_t<std::is_integral_v<T>, T> operator()(T x, T y) {
-
539 return x < y ? x : y;
-
540 }
+
533};
-
541
-
542 template <typename T>
-
-
543 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T x, T y) {
-
544 if (std::isnan(x)) {
-
545 return x;
-
546 }
-
547 return x < y ? x : y;
-
548 }
+
534
+
+
535struct Maximum {
+
536 template <typename T>
+
+
537 std::enable_if_t<std::is_integral_v<T>, T> operator()(T x, T y) {
+
538 return (x > y) ? x : y;
+
539 }
-
549};
+
540
+
541 template <typename T>
+
+
542 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T x, T y) {
+
543 if (std::isnan(x)) {
+
544 return x;
+
545 }
+
546 return (x > y) ? x : y;
+
547 }
-
550
-
-
551struct LogAddExp {
-
552 template <typename T>
-
-
553 T operator()(T x, T y) {
-
554 constexpr float inf = std::numeric_limits<float>::infinity();
-
555 auto maxval = Maximum()(x, y);
-
556 auto minval = Minimum()(x, y);
-
557 return (minval == -inf || maxval == inf)
-
558 ? maxval
-
559 : static_cast<decltype(x)>(
-
560 maxval + std::log1p(fast_exp(minval - maxval)));
-
561 }
+
548};
-
562};
+
549
+
+
550struct Minimum {
+
551 template <typename T>
+
+
552 std::enable_if_t<std::is_integral_v<T>, T> operator()(T x, T y) {
+
553 return x < y ? x : y;
+
554 }
-
563
-
-
564struct Multiply {
-
565 template <typename T>
-
-
566 T operator()(T x, T y) {
-
567 return x * y;
-
568 }
+
555
+
556 template <typename T>
+
+
557 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T x, T y) {
+
558 if (std::isnan(x)) {
+
559 return x;
+
560 }
+
561 return x < y ? x : y;
+
562 }
-
569};
+
563};
-
570
-
-
571struct NotEqual {
-
572 template <typename T>
-
-
573 bool operator()(T x, T y) {
-
574 return x != y;
+
564
+
+
565struct LogAddExp {
+
566 template <typename T>
+
+
567 T operator()(T x, T y) {
+
568 constexpr float inf = std::numeric_limits<float>::infinity();
+
569 auto maxval = Maximum()(x, y);
+
570 auto minval = Minimum()(x, y);
+
571 return (minval == -inf || maxval == inf)
+
572 ? maxval
+
573 : static_cast<decltype(x)>(
+
574 maxval + std::log1p(fast_exp(minval - maxval)));
575 }
576};
577
-
578struct Power {
+
578struct Multiply {
579 template <typename T>
-
580 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T base, T exp) {
-
581 return std::pow(base, exp);
+
580 T operator()(T x, T y) {
+
581 return x * y;
582 }
-
583
-
584 template <typename T>
-
-
585 std::enable_if_t<std::is_integral_v<T>, T> operator()(T base, T exp) {
-
586 T res = 1;
-
587 while (exp) {
-
588 if (exp & 1) {
-
589 res *= base;
-
590 }
-
591 exp >>= 1;
-
592 base *= base;
-
593 }
-
594 return res;
-
595 }
+
583};
-
596};
+
584
+
+
585struct NotEqual {
+
586 template <typename T>
+
+
587 bool operator()(T x, T y) {
+
588 return x != y;
+
589 }
+
+
590};
+
+
591
+
+
592struct Power {
+
593 template <typename T>
+
+
594 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T base, T exp) {
+
595 return std::pow(base, exp);
+
596 }
597
-
-
598struct Subtract {
-
599 template <typename T>
-
-
600 T operator()(T x, T y) {
-
601 return x - y;
-
602 }
-
-
603};
-
-
604
-
-
605struct LogicalAnd {
-
606 template <typename T>
-
-
607 T operator()(T x, T y) {
-
608 return x && y;
+
598 template <typename T>
+
+
599 std::enable_if_t<std::is_integral_v<T>, T> operator()(T base, T exp) {
+
600 T res = 1;
+
601 while (exp) {
+
602 if (exp & 1) {
+
603 res *= base;
+
604 }
+
605 exp >>= 1;
+
606 base *= base;
+
607 }
+
608 return res;
609 }
610};
611
-
612struct LogicalOr {
+
612struct Subtract {
613 template <typename T>
-
614 T operator()(T x, T y) {
-
615 return x || y;
+
614 T operator()(T x, T y) {
+
615 return x - y;
616 }
617};
618
-
619struct Select {
+
619struct LogicalAnd {
620 template <typename T>
-
621 T operator()(bool condition, T x, T y) {
-
622 return condition ? x : y;
+
621 T operator()(T x, T y) {
+
622 return x && y;
623 }
624};
625
-
626struct BitwiseAnd {
+
626struct LogicalOr {
627 template <typename T>
-
628 T operator()(T x, T y) {
-
629 return x & y;
+
628 T operator()(T x, T y) {
+
629 return x || y;
630 }
631};
632
-
633struct BitwiseOr {
+
633struct Select {
634 template <typename T>
-
635 T operator()(T x, T y) {
-
636 return x | y;
+
635 T operator()(bool condition, T x, T y) {
+
636 return condition ? x : y;
637 }
638};
639
-
640struct BitwiseXor {
+
640struct BitwiseAnd {
641 template <typename T>
-
642 T operator()(T x, T y) {
-
643 return x ^ y;
+
642 T operator()(T x, T y) {
+
643 return x & y;
644 }
645};
646
-
647struct LeftShift {
+
647struct BitwiseOr {
648 template <typename T>
-
649 T operator()(T x, T y) {
-
650 return x << y;
+
649 T operator()(T x, T y) {
+
650 return x | y;
651 }
652};
653
-
654struct RightShift {
+
654struct BitwiseXor {
655 template <typename T>
-
656 T operator()(T x, T y) {
-
657 return x >> y;
+
656 T operator()(T x, T y) {
+
657 return x ^ y;
658 }
659};
660
-
661} // namespace mlx::core::detail
+
+
661struct LeftShift {
+
662 template <typename T>
+
+
663 T operator()(T x, T y) {
+
664 return x << y;
+
665 }
+
+
666};
+
+
667
+
+
668struct RightShift {
+
669 template <typename T>
+
+
670 T operator()(T x, T y) {
+
671 return x >> y;
+
672 }
+
+
673};
+
+
674
+
675} // namespace mlx::core::detail
mlx::core::log1p
array log1p(const array &a, StreamOrDevice s={})
Natural logarithm of one plus elements in the array: log(1 + a).
mlx::core::expm1
array expm1(const array &a, StreamOrDevice s={})
Computes the expm1 function of the elements of an array.
@@ -1077,8 +1099,8 @@ $(function(){ initResizable(false); });
mlx::core::detail::Abs::operator()
uint16_t operator()(uint16_t x)
Definition ops.h:115
mlx::core::detail::Abs::operator()
bool operator()(bool x)
Definition ops.h:124
mlx::core::detail::Abs::operator()
uint32_t operator()(uint32_t x)
Definition ops.h:118
-
mlx::core::detail::Add
Definition ops.h:431
-
mlx::core::detail::Add::operator()
T operator()(T x, T y)
Definition ops.h:433
+
mlx::core::detail::Add
Definition ops.h:445
+
mlx::core::detail::Add::operator()
T operator()(T x, T y)
Definition ops.h:447
mlx::core::detail::ArcCos
Definition ops.h:129
mlx::core::detail::ArcCos::operator()
T operator()(T x)
Definition ops.h:131
mlx::core::detail::ArcCosh
Definition ops.h:136
@@ -1093,12 +1115,12 @@ $(function(){ initResizable(false); });
mlx::core::detail::ArcTan::operator()
T operator()(T x)
Definition ops.h:159
mlx::core::detail::ArcTanh
Definition ops.h:171
mlx::core::detail::ArcTanh::operator()
T operator()(T x)
Definition ops.h:173
-
mlx::core::detail::BitwiseAnd
Definition ops.h:626
-
mlx::core::detail::BitwiseAnd::operator()
T operator()(T x, T y)
Definition ops.h:628
-
mlx::core::detail::BitwiseOr
Definition ops.h:633
-
mlx::core::detail::BitwiseOr::operator()
T operator()(T x, T y)
Definition ops.h:635
-
mlx::core::detail::BitwiseXor
Definition ops.h:640
-
mlx::core::detail::BitwiseXor::operator()
T operator()(T x, T y)
Definition ops.h:642
+
mlx::core::detail::BitwiseAnd
Definition ops.h:640
+
mlx::core::detail::BitwiseAnd::operator()
T operator()(T x, T y)
Definition ops.h:642
+
mlx::core::detail::BitwiseOr
Definition ops.h:647
+
mlx::core::detail::BitwiseOr::operator()
T operator()(T x, T y)
Definition ops.h:649
+
mlx::core::detail::BitwiseXor
Definition ops.h:654
+
mlx::core::detail::BitwiseXor::operator()
T operator()(T x, T y)
Definition ops.h:656
mlx::core::detail::Ceil
Definition ops.h:178
mlx::core::detail::Ceil::operator()
uint8_t operator()(uint8_t x)
Definition ops.h:195
mlx::core::detail::Ceil::operator()
T operator()(T x)
Definition ops.h:180
@@ -1116,10 +1138,10 @@ $(function(){ initResizable(false); });
mlx::core::detail::Cos::operator()
T operator()(T x)
Definition ops.h:220
mlx::core::detail::Cosh
Definition ops.h:225
mlx::core::detail::Cosh::operator()
T operator()(T x)
Definition ops.h:227
-
mlx::core::detail::Divide
Definition ops.h:438
-
mlx::core::detail::Divide::operator()
T operator()(T x, T y)
Definition ops.h:440
-
mlx::core::detail::Equal
Definition ops.h:479
-
mlx::core::detail::Equal::operator()
bool operator()(T x, T y)
Definition ops.h:481
+
mlx::core::detail::Divide
Definition ops.h:452
+
mlx::core::detail::Divide::operator()
T operator()(T x, T y)
Definition ops.h:454
+
mlx::core::detail::Equal
Definition ops.h:493
+
mlx::core::detail::Equal::operator()
bool operator()(T x, T y)
Definition ops.h:495
mlx::core::detail::Erf
Definition ops.h:232
mlx::core::detail::Erf::operator()
T operator()(T x)
Definition ops.h:234
mlx::core::detail::ErfInv
Definition ops.h:239
@@ -1140,86 +1162,90 @@ $(function(){ initResizable(false); });
mlx::core::detail::Floor::operator()
int8_t operator()(int8_t x)
Definition ops.h:269
mlx::core::detail::Floor::operator()
uint64_t operator()(uint64_t x)
Definition ops.h:290
mlx::core::detail::Floor::operator()
int16_t operator()(int16_t x)
Definition ops.h:272
- -
mlx::core::detail::GreaterEqual::operator()
bool operator()(T x, T y)
Definition ops.h:502
-
mlx::core::detail::Greater
Definition ops.h:493
-
mlx::core::detail::Greater::operator()
bool operator()(T x, T y)
Definition ops.h:495
-
mlx::core::detail::LeftShift
Definition ops.h:647
-
mlx::core::detail::LeftShift::operator()
T operator()(T x, T y)
Definition ops.h:649
-
mlx::core::detail::LessEqual
Definition ops.h:514
-
mlx::core::detail::LessEqual::operator()
bool operator()(T x, T y)
Definition ops.h:516
-
mlx::core::detail::Less
Definition ops.h:507
-
mlx::core::detail::Less::operator()
bool operator()(T x, T y)
Definition ops.h:509
-
mlx::core::detail::Log10
Definition ops.h:312
-
mlx::core::detail::Log10::operator()
T operator()(T x)
Definition ops.h:314
-
mlx::core::detail::Log1p
Definition ops.h:319
-
mlx::core::detail::Log1p::operator()
T operator()(T x)
Definition ops.h:321
-
mlx::core::detail::Log2
Definition ops.h:305
-
mlx::core::detail::Log2::operator()
T operator()(T x)
Definition ops.h:307
-
mlx::core::detail::LogAddExp
Definition ops.h:551
-
mlx::core::detail::LogAddExp::operator()
T operator()(T x, T y)
Definition ops.h:553
-
mlx::core::detail::Log
Definition ops.h:298
-
mlx::core::detail::Log::operator()
T operator()(T x)
Definition ops.h:300
-
mlx::core::detail::LogicalAnd
Definition ops.h:605
-
mlx::core::detail::LogicalAnd::operator()
T operator()(T x, T y)
Definition ops.h:607
-
mlx::core::detail::LogicalNot
Definition ops.h:326
-
mlx::core::detail::LogicalNot::operator()
T operator()(T x)
Definition ops.h:328
-
mlx::core::detail::LogicalOr
Definition ops.h:612
-
mlx::core::detail::LogicalOr::operator()
T operator()(T x, T y)
Definition ops.h:614
-
mlx::core::detail::Maximum
Definition ops.h:521
-
mlx::core::detail::Maximum::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:523
-
mlx::core::detail::Maximum::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:528
-
mlx::core::detail::Minimum
Definition ops.h:536
-
mlx::core::detail::Minimum::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:543
-
mlx::core::detail::Minimum::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:538
-
mlx::core::detail::Multiply
Definition ops.h:564
-
mlx::core::detail::Multiply::operator()
T operator()(T x, T y)
Definition ops.h:566
-
mlx::core::detail::NaNEqual
Definition ops.h:486
-
mlx::core::detail::NaNEqual::operator()
bool operator()(T x, T y)
Definition ops.h:488
-
mlx::core::detail::Negative
Definition ops.h:333
-
mlx::core::detail::Negative::operator()
T operator()(T x)
Definition ops.h:335
-
mlx::core::detail::NotEqual
Definition ops.h:571
-
mlx::core::detail::NotEqual::operator()
bool operator()(T x, T y)
Definition ops.h:573
-
mlx::core::detail::Power
Definition ops.h:578
-
mlx::core::detail::Power::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T base, T exp)
Definition ops.h:580
-
mlx::core::detail::Power::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T base, T exp)
Definition ops.h:585
-
mlx::core::detail::Remainder
Definition ops.h:445
-
mlx::core::detail::Remainder::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:464
-
mlx::core::detail::Remainder::operator()
std::enable_if_t< std::is_integral_v< T > &!std::is_signed_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:447
-
mlx::core::detail::Remainder::operator()
std::enable_if_t< std::is_integral_v< T > &std::is_signed_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:454
-
mlx::core::detail::Remainder::operator()
complex64_t operator()(complex64_t numerator, complex64_t denominator)
Definition ops.h:474
-
mlx::core::detail::RightShift
Definition ops.h:654
-
mlx::core::detail::RightShift::operator()
T operator()(T x, T y)
Definition ops.h:656
-
mlx::core::detail::Round
Definition ops.h:340
-
mlx::core::detail::Round::operator()
T operator()(T x)
Definition ops.h:342
-
mlx::core::detail::Round::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:346
-
mlx::core::detail::Rsqrt
Definition ops.h:410
-
mlx::core::detail::Rsqrt::operator()
T operator()(T x)
Definition ops.h:412
-
mlx::core::detail::Select
Definition ops.h:619
-
mlx::core::detail::Select::operator()
T operator()(bool condition, T x, T y)
Definition ops.h:621
-
mlx::core::detail::Sigmoid
Definition ops.h:351
-
mlx::core::detail::Sigmoid::operator()
T operator()(T x)
Definition ops.h:353
-
mlx::core::detail::Sign
Definition ops.h:359
-
mlx::core::detail::Sign::operator()
uint64_t operator()(uint64_t x)
Definition ops.h:373
-
mlx::core::detail::Sign::operator()
T operator()(T x)
Definition ops.h:361
-
mlx::core::detail::Sign::operator()
uint8_t operator()(uint8_t x)
Definition ops.h:364
-
mlx::core::detail::Sign::operator()
uint16_t operator()(uint16_t x)
Definition ops.h:367
-
mlx::core::detail::Sign::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:377
-
mlx::core::detail::Sign::operator()
uint32_t operator()(uint32_t x)
Definition ops.h:370
-
mlx::core::detail::Sin
Definition ops.h:382
-
mlx::core::detail::Sin::operator()
T operator()(T x)
Definition ops.h:384
-
mlx::core::detail::Sinh
Definition ops.h:389
-
mlx::core::detail::Sinh::operator()
T operator()(T x)
Definition ops.h:391
-
mlx::core::detail::Sqrt
Definition ops.h:403
-
mlx::core::detail::Sqrt::operator()
T operator()(T x)
Definition ops.h:405
-
mlx::core::detail::Square
Definition ops.h:396
-
mlx::core::detail::Square::operator()
T operator()(T x)
Definition ops.h:398
-
mlx::core::detail::Subtract
Definition ops.h:598
-
mlx::core::detail::Subtract::operator()
T operator()(T x, T y)
Definition ops.h:600
-
mlx::core::detail::Tan
Definition ops.h:417
-
mlx::core::detail::Tan::operator()
T operator()(T x)
Definition ops.h:419
-
mlx::core::detail::Tanh
Definition ops.h:424
-
mlx::core::detail::Tanh::operator()
T operator()(T x)
Definition ops.h:426
+ +
mlx::core::detail::GreaterEqual::operator()
bool operator()(T x, T y)
Definition ops.h:516
+
mlx::core::detail::Greater
Definition ops.h:507
+
mlx::core::detail::Greater::operator()
bool operator()(T x, T y)
Definition ops.h:509
+
mlx::core::detail::Imag
Definition ops.h:298
+
mlx::core::detail::Imag::operator()
T operator()(T x)
Definition ops.h:300
+
mlx::core::detail::LeftShift
Definition ops.h:661
+
mlx::core::detail::LeftShift::operator()
T operator()(T x, T y)
Definition ops.h:663
+
mlx::core::detail::LessEqual
Definition ops.h:528
+
mlx::core::detail::LessEqual::operator()
bool operator()(T x, T y)
Definition ops.h:530
+
mlx::core::detail::Less
Definition ops.h:521
+
mlx::core::detail::Less::operator()
bool operator()(T x, T y)
Definition ops.h:523
+
mlx::core::detail::Log10
Definition ops.h:319
+
mlx::core::detail::Log10::operator()
T operator()(T x)
Definition ops.h:321
+
mlx::core::detail::Log1p
Definition ops.h:326
+
mlx::core::detail::Log1p::operator()
T operator()(T x)
Definition ops.h:328
+
mlx::core::detail::Log2
Definition ops.h:312
+
mlx::core::detail::Log2::operator()
T operator()(T x)
Definition ops.h:314
+
mlx::core::detail::LogAddExp
Definition ops.h:565
+
mlx::core::detail::LogAddExp::operator()
T operator()(T x, T y)
Definition ops.h:567
+
mlx::core::detail::Log
Definition ops.h:305
+
mlx::core::detail::Log::operator()
T operator()(T x)
Definition ops.h:307
+
mlx::core::detail::LogicalAnd
Definition ops.h:619
+
mlx::core::detail::LogicalAnd::operator()
T operator()(T x, T y)
Definition ops.h:621
+
mlx::core::detail::LogicalNot
Definition ops.h:333
+
mlx::core::detail::LogicalNot::operator()
T operator()(T x)
Definition ops.h:335
+
mlx::core::detail::LogicalOr
Definition ops.h:626
+
mlx::core::detail::LogicalOr::operator()
T operator()(T x, T y)
Definition ops.h:628
+
mlx::core::detail::Maximum
Definition ops.h:535
+
mlx::core::detail::Maximum::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:537
+
mlx::core::detail::Maximum::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:542
+
mlx::core::detail::Minimum
Definition ops.h:550
+
mlx::core::detail::Minimum::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:557
+
mlx::core::detail::Minimum::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:552
+
mlx::core::detail::Multiply
Definition ops.h:578
+
mlx::core::detail::Multiply::operator()
T operator()(T x, T y)
Definition ops.h:580
+
mlx::core::detail::NaNEqual
Definition ops.h:500
+
mlx::core::detail::NaNEqual::operator()
bool operator()(T x, T y)
Definition ops.h:502
+
mlx::core::detail::Negative
Definition ops.h:340
+
mlx::core::detail::Negative::operator()
T operator()(T x)
Definition ops.h:342
+
mlx::core::detail::NotEqual
Definition ops.h:585
+
mlx::core::detail::NotEqual::operator()
bool operator()(T x, T y)
Definition ops.h:587
+
mlx::core::detail::Power
Definition ops.h:592
+
mlx::core::detail::Power::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T base, T exp)
Definition ops.h:594
+
mlx::core::detail::Power::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T base, T exp)
Definition ops.h:599
+
mlx::core::detail::Real
Definition ops.h:347
+
mlx::core::detail::Real::operator()
T operator()(T x)
Definition ops.h:349
+
mlx::core::detail::Remainder
Definition ops.h:459
+
mlx::core::detail::Remainder::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:478
+
mlx::core::detail::Remainder::operator()
std::enable_if_t< std::is_integral_v< T > &!std::is_signed_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:461
+
mlx::core::detail::Remainder::operator()
std::enable_if_t< std::is_integral_v< T > &std::is_signed_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:468
+
mlx::core::detail::Remainder::operator()
complex64_t operator()(complex64_t numerator, complex64_t denominator)
Definition ops.h:488
+
mlx::core::detail::RightShift
Definition ops.h:668
+
mlx::core::detail::RightShift::operator()
T operator()(T x, T y)
Definition ops.h:670
+
mlx::core::detail::Round
Definition ops.h:354
+
mlx::core::detail::Round::operator()
T operator()(T x)
Definition ops.h:356
+
mlx::core::detail::Round::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:360
+
mlx::core::detail::Rsqrt
Definition ops.h:424
+
mlx::core::detail::Rsqrt::operator()
T operator()(T x)
Definition ops.h:426
+
mlx::core::detail::Select
Definition ops.h:633
+
mlx::core::detail::Select::operator()
T operator()(bool condition, T x, T y)
Definition ops.h:635
+
mlx::core::detail::Sigmoid
Definition ops.h:365
+
mlx::core::detail::Sigmoid::operator()
T operator()(T x)
Definition ops.h:367
+
mlx::core::detail::Sign
Definition ops.h:373
+
mlx::core::detail::Sign::operator()
uint64_t operator()(uint64_t x)
Definition ops.h:387
+
mlx::core::detail::Sign::operator()
T operator()(T x)
Definition ops.h:375
+
mlx::core::detail::Sign::operator()
uint8_t operator()(uint8_t x)
Definition ops.h:378
+
mlx::core::detail::Sign::operator()
uint16_t operator()(uint16_t x)
Definition ops.h:381
+
mlx::core::detail::Sign::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:391
+
mlx::core::detail::Sign::operator()
uint32_t operator()(uint32_t x)
Definition ops.h:384
+
mlx::core::detail::Sin
Definition ops.h:396
+
mlx::core::detail::Sin::operator()
T operator()(T x)
Definition ops.h:398
+
mlx::core::detail::Sinh
Definition ops.h:403
+
mlx::core::detail::Sinh::operator()
T operator()(T x)
Definition ops.h:405
+
mlx::core::detail::Sqrt
Definition ops.h:417
+
mlx::core::detail::Sqrt::operator()
T operator()(T x)
Definition ops.h:419
+
mlx::core::detail::Square
Definition ops.h:410
+
mlx::core::detail::Square::operator()
T operator()(T x)
Definition ops.h:412
+
mlx::core::detail::Subtract
Definition ops.h:612
+
mlx::core::detail::Subtract::operator()
T operator()(T x, T y)
Definition ops.h:614
+
mlx::core::detail::Tan
Definition ops.h:431
+
mlx::core::detail::Tan::operator()
T operator()(T x)
Definition ops.h:433
+
mlx::core::detail::Tanh
Definition ops.h:438
+
mlx::core::detail::Tanh::operator()
T operator()(T x)
Definition ops.h:440
u
uint32_t u
Definition bf16.h:17
mlx::core::detail::IntOrFloat::f
float f
Definition ops.h:16
diff --git a/docs/build/html/backend_2metal_2kernels_2fft_8h_source.html b/docs/build/html/backend_2metal_2kernels_2fft_8h_source.html index 0f30f51c8..a64fb6ae8 100644 --- a/docs/build/html/backend_2metal_2kernels_2fft_8h_source.html +++ b/docs/build/html/backend_2metal_2kernels_2fft_8h_source.html @@ -546,7 +546,7 @@ $(function(){ initResizable(false); });
439 typename in_T,
440 typename out_T,
441 int step,
-
442 bool real = false>
+
442 bool real = false>
443[[kernel]] void four_step_fft(
444 const device in_T* in [[buffer(0)]],
@@ -627,6 +627,7 @@ $(function(){ initResizable(false); });
elems_per_thread_
static constant constexpr const int elems_per_thread_
Definition fft.h:25
rader_3_steps_
static constant constexpr const int rader_3_steps_
Definition fft.h:46
rader_5_steps_
static constant constexpr const int rader_5_steps_
Definition fft.h:44
+
mlx::core::real
array real(const array &a, StreamOrDevice s={})
radix_func
METAL_FUNC void radix_func(thread float *x)
Definition hadamard.h:11
metal
Definition bf16.h:265
metal::min
METAL_FUNC bfloat16_t min(bfloat16_t x, bfloat16_t y)
Definition bf16_math.h:234
diff --git a/docs/build/html/backend_2metal_2utils_8h.html b/docs/build/html/backend_2metal_2utils_8h.html index 1f22ee294..d60c91e3d 100644 --- a/docs/build/html/backend_2metal_2utils_8h.html +++ b/docs/build/html/backend_2metal_2utils_8h.html @@ -117,8 +117,8 @@ Functions   std::string mlx::core::type_to_name (const array &a)   -MTL::Size mlx::core::get_block_dims (int dim0, int dim1, int dim2) -  +MTL::Size mlx::core::get_block_dims (int dim0, int dim1, int dim2, int pow2=10) +  MTL::Size mlx::core::get_2d_grid_dims (const std::vector< int > &shape, const std::vector< size_t > &strides)   NS::String * mlx::core::make_string (std::ostringstream &os) diff --git a/docs/build/html/backend_2metal_2utils_8h_source.html b/docs/build/html/backend_2metal_2utils_8h_source.html index 4a0dc3bfd..0c09faa49 100644 --- a/docs/build/html/backend_2metal_2utils_8h_source.html +++ b/docs/build/html/backend_2metal_2utils_8h_source.html @@ -127,8 +127,8 @@ $(function(){ initResizable(false); });
30// Compute the thread block dimensions which fit the given
31// input dimensions.
32// - The thread block dimensions will be powers of two
-
33// - The thread block size will be less than 1024
-
34MTL::Size get_block_dims(int dim0, int dim1, int dim2);
+
33// - The thread block size will be less than 2^pow2
+
34MTL::Size get_block_dims(int dim0, int dim1, int dim2, int pow2 = 10);
35
36// Computes a 2D grid where each element is < UINT_MAX
37// Assumes:
@@ -180,10 +180,10 @@ $(function(){ initResizable(false); });
mlx::core::Primitive::print
virtual void print(std::ostream &os)=0
Print the primitive.
mlx::core::array
Definition array.h:20
mlx::core
Definition allocator.h:7
+
mlx::core::get_block_dims
MTL::Size get_block_dims(int dim0, int dim1, int dim2, int pow2=10)
mlx::core::debug_set_primitive_buffer_label
void debug_set_primitive_buffer_label(MTL::CommandBuffer *command_buffer, Primitive &primitive)
Definition utils.h:58
mlx::core::set_vector_bytes
void set_vector_bytes(CommandEncoder &enc, const std::vector< T > &vec, size_t nelems, int idx)
Definition utils.h:14
mlx::core::debug_set_stream_queue_label
void debug_set_stream_queue_label(MTL::CommandQueue *queue, int index)
Definition utils.h:50
-
mlx::core::get_block_dims
MTL::Size get_block_dims(int dim0, int dim1, int dim2)
mlx::core::get_2d_grid_dims
MTL::Size get_2d_grid_dims(const std::vector< int > &shape, const std::vector< size_t > &strides)
mlx::core::get_primitive_string
std::string get_primitive_string(Primitive *primitive)
mlx::core::make_string
NS::String * make_string(std::ostringstream &os)
Definition utils.h:45
diff --git a/docs/build/html/classes.html b/docs/build/html/classes.html index c3d12f56d..45cdd6132 100644 --- a/docs/build/html/classes.html +++ b/docs/build/html/classes.html @@ -115,7 +115,7 @@ $(function(){ initResizable(false); });
Hadamard (mlx::core)
I
-
ImplicitGemmConv2DParams (mlx::steel)
Indices
IntOrFloat (mlx::core::detail)
InTracing (mlx::core::detail)
Inverse (mlx::core)
+
Imag
Imag (mlx::core::detail)
Imag (mlx::core)
ImplicitGemmConv2DParams (mlx::steel)
Indices
IntOrFloat (mlx::core::detail)
InTracing (mlx::core::detail)
Inverse (mlx::core)
K
KernelMergeSort
KernelMultiBlockMergeSort
KeySequence (mlx::core::random)
@@ -139,7 +139,7 @@ $(function(){ initResizable(false); });
QRF (mlx::core)
QuantizedBlockLoader
QuantizedMatmul (mlx::core)
R
-
RandomBits (mlx::core)
Reader (mlx::core::io)
BlockLoader::ReadVector (mlx::steel)
ReadWriter
Recv (mlx::core::distributed)
Reduce (mlx::core)
ReductionPlan (mlx::core)
Remainder (mlx::core::detail)
Remainder (mlx::core)
Remainder
Reshape (mlx::core)
RetainGraph (mlx::core::detail)
rev_iter (pocketfft::detail)
rfftp (pocketfft::detail)
RightShift (mlx::core::detail)
RightShift
RMSNorm (mlx::core::fast)
RMSNormVJP (mlx::core::fast)
RoPE (mlx::core::fast)
Round (mlx::core::detail)
Round (mlx::core)
Round
Rsqrt (mlx::core::detail)
Rsqrt
+
RandomBits (mlx::core)
Reader (mlx::core::io)
BlockLoader::ReadVector (mlx::steel)
ReadWriter
Real (mlx::core::detail)
Real (mlx::core)
Real
Recv (mlx::core::distributed)
Reduce (mlx::core)
ReductionPlan (mlx::core)
Remainder (mlx::core::detail)
Remainder (mlx::core)
Remainder
Reshape (mlx::core)
RetainGraph (mlx::core::detail)
rev_iter (pocketfft::detail)
rfftp (pocketfft::detail)
RightShift (mlx::core::detail)
RightShift
RMSNorm (mlx::core::fast)
RMSNormVJP (mlx::core::fast)
RoPE (mlx::core::fast)
Round (mlx::core::detail)
Round (mlx::core)
Round
Rsqrt (mlx::core::detail)
Rsqrt
S
ScaledDotProductAttention (mlx::core::fast)
ScaleOp
Scan (mlx::core)
Scatter (mlx::core)
Scheduler (mlx::core::scheduler)
Select (mlx::core::detail)
Select (mlx::core)
Select
Send (mlx::core::distributed)
Sigmoid (mlx::core::detail)
Sigmoid (mlx::core)
Sigmoid
Sign (mlx::core::detail)
Sign (mlx::core)
Sign
simple_iter (pocketfft::detail)
Sin (mlx::core::detail)
Sin (mlx::core)
Sin
sincos_2pibyn (pocketfft::detail)
Sinh (mlx::core::detail)
Sinh (mlx::core)
Sinh
Slice (mlx::core)
SliceUpdate (mlx::core)
Softmax (mlx::core)
Sort (mlx::core)
Split (mlx::core)
Sqrt (mlx::core::detail)
Sqrt (mlx::core)
Sqrt
Square (mlx::core::detail)
Square (mlx::core)
Square
StopGradient (mlx::core)
Stream (mlx::core)
StreamContext (mlx::core)
StreamThread (mlx::core::scheduler)
Subtract (mlx::core::detail)
Subtract (mlx::core)
Subtract
Sum
SVD (mlx::core)
diff --git a/docs/build/html/classmlx_1_1core_1_1_f_f_t.html b/docs/build/html/classmlx_1_1core_1_1_f_f_t.html index 03632f818..15cbb62c2 100644 --- a/docs/build/html/classmlx_1_1core_1_1_f_f_t.html +++ b/docs/build/html/classmlx_1_1core_1_1_f_f_t.html @@ -109,7 +109,7 @@ Inheritance diagram for mlx::core::FFT:
- + diff --git a/docs/build/html/classmlx_1_1core_1_1_imag-members.html b/docs/build/html/classmlx_1_1core_1_1_imag-members.html new file mode 100644 index 000000000..b03925f76 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_imag-members.html @@ -0,0 +1,129 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + +
+
+

Public Member Functions

 FFT (Stream stream, const std::vector< size_t > &axes, bool inverse, bool real)
 FFT (Stream stream, const std::vector< size_t > &axes, bool inverse, bool real)
 
void eval_cpu (const std::vector< array > &inputs, array &out) override
 
+ + + + + +
+
MLX +
+
+
+ + + + + + + + + + +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ + +
+
+
+
mlx::core::Imag Member List
+
+
+ +

This is the complete list of members for mlx::core::Imag, including all inherited members.

+ + + + + + + + + + + + + + + + + + + + + + + + + + +
device()mlx::core::Primitiveinline
eval_cpu(const std::vector< array > &inputs, array &out) overridemlx::core::Imagvirtual
mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
eval_gpu(const std::vector< array > &inputs, array &out) overridemlx::core::Imagvirtual
mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
Imag(Stream stream)mlx::core::Imaginlineexplicit
is_equivalent(const Primitive &other) const overridemlx::core::Imaginlinevirtual
jvp(const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) overridemlx::core::Imagvirtual
operator=(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
operator=(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
mlx::core::Primitive::operator=(const Primitive &other)=deletemlx::core::Primitive
mlx::core::Primitive::operator=(Primitive &&other)=deletemlx::core::Primitive
output_shapes(const std::vector< array > &inputs) overridemlx::core::Imaginlinevirtual
Primitive(Stream stream)mlx::core::Primitiveinlineexplicit
Primitive(const Primitive &other)=deletemlx::core::Primitive
Primitive(Primitive &&other)=deletemlx::core::Primitive
print(std::ostream &os) overridemlx::core::Imaginlinevirtual
stream()mlx::core::Primitiveinline
UnaryPrimitive(Stream stream)mlx::core::UnaryPrimitiveinlineexplicit
UnaryPrimitive(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
UnaryPrimitive(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
vjp(const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) overridemlx::core::Imagvirtual
vmap(const std::vector< array > &inputs, const std::vector< int > &axes) overridemlx::core::Imagvirtual
~Primitive()=defaultmlx::core::Primitivevirtual
~UnaryPrimitive()=defaultmlx::core::UnaryPrimitivevirtual
+ +
+Generated by doxygen 1.12.0 +
+
+ + diff --git a/docs/build/html/classmlx_1_1core_1_1_imag.html b/docs/build/html/classmlx_1_1core_1_1_imag.html new file mode 100644 index 000000000..4b9c3bd78 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_imag.html @@ -0,0 +1,477 @@ + + + + + + + +MLX: mlx::core::Imag Class Reference + + + + + + + + + + + + + +
+
+ + + + + + +
+
MLX +
+
+
+ + + + + + + + + + +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ + +
+
+
+
+Public Member Functions | +List of all members
+
mlx::core::Imag Class Reference
+
+
+ +

#include <primitives.h>

+
+Inheritance diagram for mlx::core::Imag:
+
+
+ + +mlx::core::UnaryPrimitive +mlx::core::Primitive + +
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 Imag (Stream stream)
 
void eval_cpu (const std::vector< array > &inputs, array &out) override
 
void eval_gpu (const std::vector< array > &inputs, array &out) override
 
virtual std::pair< std::vector< array >, std::vector< int > > vmap (const std::vector< array > &inputs, const std::vector< int > &axes) override
 The primitive must know how to vectorize itself across the given axes.
 
std::vector< array > jvp (const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) override
 The Jacobian-vector product.
 
std::vector< array > vjp (const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) override
 The vector-Jacobian product.
 
void print (std::ostream &os) override
 Print the primitive.
 
bool is_equivalent (const Primitive &other) const override
 Equivalence check defaults to false unless overridden by the primitive.
 
std::vector< std::vector< int > > output_shapes (const std::vector< array > &inputs) override
 Get the output shapes of the primitive.
 
- Public Member Functions inherited from mlx::core::UnaryPrimitive
 UnaryPrimitive (Stream stream)
 An abstract base class for a primitive with a single output.
 
void eval_cpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the output arrays.
 
void eval_gpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 
virtual ~UnaryPrimitive ()=default
 
 UnaryPrimitive (const UnaryPrimitive &other)=delete
 
 UnaryPrimitive (UnaryPrimitive &&other)=delete
 
UnaryPrimitive & operator= (const UnaryPrimitive &other)=delete
 
UnaryPrimitive & operator= (UnaryPrimitive &&other)=delete
 
- Public Member Functions inherited from mlx::core::Primitive
 Primitive (Stream stream)
 
const Device & device ()
 The device the primitive will run on.
 
const Stream & stream ()
 The stream the primitive will run on.
 
virtual ~Primitive ()=default
 
 Primitive (const Primitive &other)=delete
 
 Primitive (Primitive &&other)=delete
 
Primitive & operator= (const Primitive &other)=delete
 
Primitive & operator= (Primitive &&other)=delete
 
+

Constructor & Destructor Documentation

+ +

◆ Imag()

+ +
+
+ + + + + +
+ + + + + + + +
mlx::core::Imag::Imag (Stream stream)
+ 		+inlineexplicit
+
+ +
+
+

Member Function Documentation

+ +

◆ eval_cpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::Imag::eval_cpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ eval_gpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::Imag::eval_gpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ is_equivalent()

+ +
+
+ + + + + +
+ + + + + + + +
bool mlx::core::Imag::is_equivalent (const Primitive & other) const
+ 		+inlineoverridevirtual
+
+ +

Equivalence check defaults to false unless overridden by the primitive.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ jvp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::Imag::jvp (const std::vector< array > & primals,
const std::vector< array > & tangents,
const std::vector< int > & argnums )
+ 		+overridevirtual
+
+ +

The Jacobian-vector product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ output_shapes()

+ +
+
+ + + + + +
+ + + + + + + +
std::vector< std::vector< int > > mlx::core::Imag::output_shapes (const std::vector< array > & inputs)
+ 		+inlineoverridevirtual
+
+ +

Get the output shapes of the primitive.

+

This is not required to be implemented by derived classes, in which case it will throw.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ print()

+ +
+
+ + + + + +
+ + + + + + + +
void mlx::core::Imag::print (std::ostream & os)
+ 		+inlineoverridevirtual
+
+ +

Print the primitive.

+ +

Implements mlx::core::Primitive.

+ +
+
+ +

◆ vjp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::Imag::vjp (const std::vector< array > & primals,
const std::vector< array > & cotangents,
const std::vector< int > & argnums,
const std::vector< array > & outputs )
+ 		+overridevirtual
+
+ +

The vector-Jacobian product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ vmap()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
virtual std::pair< std::vector< array >, std::vector< int > > mlx::core::Imag::vmap (const std::vector< array > & inputs,
const std::vector< int > & axes )
+ 		+overridevirtual
+
+ +

The primitive must know how to vectorize itself across the given axes.

+

The output is a pair containing the output arrays representing the vectorized computation and the axes which corresponds to the vectorized dimensions of each output.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+
The documentation for this class was generated from the following file: +
+ +
+Generated by doxygen 1.12.0 +
+
+ + diff --git a/docs/build/html/classmlx_1_1core_1_1_imag.png b/docs/build/html/classmlx_1_1core_1_1_imag.png new file mode 100644 index 000000000..47ce12655 Binary files /dev/null and b/docs/build/html/classmlx_1_1core_1_1_imag.png differ diff --git a/docs/build/html/classmlx_1_1core_1_1_primitive.html b/docs/build/html/classmlx_1_1core_1_1_primitive.html index f0e744977..84e00c111 100644 --- a/docs/build/html/classmlx_1_1core_1_1_primitive.html +++ b/docs/build/html/classmlx_1_1core_1_1_primitive.html @@ -379,7 +379,7 @@ Public Member Functions

Equivalence check defaults to false unless overridden by the primitive.

-

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

+

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

@@ -417,7 +417,7 @@ Public Member Functions

The Jacobian-vector product.

-

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, and mlx::core::Transpose.

+

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::Real, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, and mlx::core::Transpose.

@@ -497,7 +497,7 @@ Public Member Functions

Get the output shapes of the primitive.

This is not required to be implemented by derived classes, in which case it will throw.

-

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::Floor, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Partition, mlx::core::Power, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Round, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, and mlx::core::Tanh.

+

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::Floor, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Partition, mlx::core::Power, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Round, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, and mlx::core::Tanh.

@@ -526,7 +526,7 @@ Public Member Functions

Print the primitive.

-

Implemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

+

Implemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

@@ -596,7 +596,7 @@ Public Member Functions

The vector-Jacobian product.

-

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::fast::LayerNorm, mlx::core::fast::RMSNorm, mlx::core::fast::RoPE, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, and mlx::core::Transpose.

+

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::fast::LayerNorm, mlx::core::fast::RMSNorm, mlx::core::fast::RoPE, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, and mlx::core::Transpose.

@@ -630,7 +630,7 @@ Public Member Functions

The primitive must know how to vectorize itself across the given axes.

The output is a pair containing the output arrays representing the vectorized computation and the axes which corresponds to the vectorized dimensions of each output.

-

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

+

Reimplemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

diff --git a/docs/build/html/classmlx_1_1core_1_1_real-members.html b/docs/build/html/classmlx_1_1core_1_1_real-members.html new file mode 100644 index 000000000..54ddb3223 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_real-members.html @@ -0,0 +1,129 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + +
+
+ + + + + + +
+
MLX +
+
+
+ + + + + + + + + + +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ + +
+
+
+
mlx::core::Real Member List
+
+
+ +

This is the complete list of members for mlx::core::Real, including all inherited members.

+ + + + + + + + + + + + + + + + + + + + + + + + + + +
device()mlx::core::Primitiveinline
eval_cpu(const std::vector< array > &inputs, array &out) overridemlx::core::Realvirtual
mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
eval_gpu(const std::vector< array > &inputs, array &out) overridemlx::core::Realvirtual
mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
is_equivalent(const Primitive &other) const overridemlx::core::Realinlinevirtual
jvp(const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) overridemlx::core::Realvirtual
operator=(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
operator=(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
mlx::core::Primitive::operator=(const Primitive &other)=deletemlx::core::Primitive
mlx::core::Primitive::operator=(Primitive &&other)=deletemlx::core::Primitive
output_shapes(const std::vector< array > &inputs) overridemlx::core::Realinlinevirtual
Primitive(Stream stream)mlx::core::Primitiveinlineexplicit
Primitive(const Primitive &other)=deletemlx::core::Primitive
Primitive(Primitive &&other)=deletemlx::core::Primitive
print(std::ostream &os) overridemlx::core::Realinlinevirtual
Real(Stream stream)mlx::core::Realinlineexplicit
stream()mlx::core::Primitiveinline
UnaryPrimitive(Stream stream)mlx::core::UnaryPrimitiveinlineexplicit
UnaryPrimitive(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
UnaryPrimitive(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
vjp(const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) overridemlx::core::Realvirtual
vmap(const std::vector< array > &inputs, const std::vector< int > &axes) overridemlx::core::Realvirtual
~Primitive()=defaultmlx::core::Primitivevirtual
~UnaryPrimitive()=defaultmlx::core::UnaryPrimitivevirtual
+ +
+Generated by doxygen 1.12.0 +
+
+ + diff --git a/docs/build/html/classmlx_1_1core_1_1_real.html b/docs/build/html/classmlx_1_1core_1_1_real.html new file mode 100644 index 000000000..f606b5fa0 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_real.html @@ -0,0 +1,477 @@ + + + + + + + +MLX: mlx::core::Real Class Reference + + + + + + + + + + + + + +
+
+ + + + + + +
+
MLX +
+
+
+ + + + + + + + + + +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ + +
+
+
+
+Public Member Functions | +List of all members
+
mlx::core::Real Class Reference
+
+
+ +

#include <primitives.h>

+
+Inheritance diagram for mlx::core::Real:
+
+
+ + +mlx::core::UnaryPrimitive +mlx::core::Primitive + +
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 Real (Stream stream)
 
void eval_cpu (const std::vector< array > &inputs, array &out) override
 
void eval_gpu (const std::vector< array > &inputs, array &out) override
 
virtual std::pair< std::vector< array >, std::vector< int > > vmap (const std::vector< array > &inputs, const std::vector< int > &axes) override
 The primitive must know how to vectorize itself across the given axes.
 
std::vector< array > jvp (const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) override
 The Jacobian-vector product.
 
std::vector< array > vjp (const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) override
 The vector-Jacobian product.
 
void print (std::ostream &os) override
 Print the primitive.
 
bool is_equivalent (const Primitive &other) const override
 Equivalence check defaults to false unless overridden by the primitive.
 
std::vector< std::vector< int > > output_shapes (const std::vector< array > &inputs) override
 Get the output shapes of the primitive.
 
- Public Member Functions inherited from mlx::core::UnaryPrimitive
 UnaryPrimitive (Stream stream)
 An abstract base class for a primitive with a single output.
 
void eval_cpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the output arrays.
 
void eval_gpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 
virtual ~UnaryPrimitive ()=default
 
 UnaryPrimitive (const UnaryPrimitive &other)=delete
 
 UnaryPrimitive (UnaryPrimitive &&other)=delete
 
UnaryPrimitive & operator= (const UnaryPrimitive &other)=delete
 
UnaryPrimitive & operator= (UnaryPrimitive &&other)=delete
 
- Public Member Functions inherited from mlx::core::Primitive
 Primitive (Stream stream)
 
const Device & device ()
 The device the primitive will run on.
 
const Stream & stream ()
 The stream the primitive will run on.
 
virtual ~Primitive ()=default
 
 Primitive (const Primitive &other)=delete
 
 Primitive (Primitive &&other)=delete
 
Primitive & operator= (const Primitive &other)=delete
 
Primitive & operator= (Primitive &&other)=delete
 
+

Constructor & Destructor Documentation

+ +

◆ Real()

+ +
+
+ + + + + +
+ + + + + + + +
mlx::core::Real::Real (Stream stream)
+ 		+inlineexplicit
+
+ +
+
+

Member Function Documentation

+ +

◆ eval_cpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::Real::eval_cpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ eval_gpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::Real::eval_gpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ is_equivalent()

+ +
+
+ + + + + +
+ + + + + + + +
bool mlx::core::Real::is_equivalent (const Primitive & other) const
+ 		+inlineoverridevirtual
+
+ +

Equivalence check defaults to false unless overridden by the primitive.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ jvp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::Real::jvp (const std::vector< array > & primals,
const std::vector< array > & tangents,
const std::vector< int > & argnums )
+ 		+overridevirtual
+
+ +

The Jacobian-vector product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ output_shapes()

+ +
+
+ + + + + +
+ + + + + + + +
std::vector< std::vector< int > > mlx::core::Real::output_shapes (const std::vector< array > & inputs)
+ 		+inlineoverridevirtual
+
+ +

Get the output shapes of the primitive.

+

This is not required to be implemented by derived classes, in which case it will throw.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ print()

+ +
+
+ + + + + +
+ + + + + + + +
void mlx::core::Real::print (std::ostream & os)
+ 		+inlineoverridevirtual
+
+ +

Print the primitive.

+ +

Implements mlx::core::Primitive.

+ +
+
+ +

◆ vjp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::Real::vjp (const std::vector< array > & primals,
const std::vector< array > & cotangents,
const std::vector< int > & argnums,
const std::vector< array > & outputs )
+ 		+overridevirtual
+
+ +

The vector-Jacobian product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ vmap()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
virtual std::pair< std::vector< array >, std::vector< int > > mlx::core::Real::vmap (const std::vector< array > & inputs,
const std::vector< int > & axes )
+ 		+overridevirtual
+
+ +

The primitive must know how to vectorize itself across the given axes.

+

The output is a pair containing the output arrays representing the vectorized computation and the axes which corresponds to the vectorized dimensions of each output.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+
The documentation for this class was generated from the following file: +
+ +
+Generated by doxygen 1.12.0 +
+
+ + diff --git a/docs/build/html/classmlx_1_1core_1_1_real.png b/docs/build/html/classmlx_1_1core_1_1_real.png new file mode 100644 index 000000000..2167288fd Binary files /dev/null and b/docs/build/html/classmlx_1_1core_1_1_real.png differ diff --git a/docs/build/html/classmlx_1_1core_1_1_unary_primitive.html b/docs/build/html/classmlx_1_1core_1_1_unary_primitive.html index 154849a47..8335de9db 100644 --- a/docs/build/html/classmlx_1_1core_1_1_unary_primitive.html +++ b/docs/build/html/classmlx_1_1core_1_1_unary_primitive.html @@ -145,52 +145,54 @@ Inheritance diagram for mlx::core::UnaryPrimitive:
mlx::core::Greater mlx::core::GreaterEqual mlx::core::Hadamard -mlx::core::Inverse -mlx::core::Less -mlx::core::LessEqual -mlx::core::Load -mlx::core::Log -mlx::core::Log1p -mlx::core::LogAddExp -mlx::core::LogicalAnd -mlx::core::LogicalNot -mlx::core::LogicalOr -mlx::core::Matmul -mlx::core::Maximum -mlx::core::Minimum -mlx::core::Multiply -mlx::core::Negative -mlx::core::NotEqual -mlx::core::NumberOfElements -mlx::core::Pad -mlx::core::Partition -mlx::core::Power -mlx::core::QuantizedMatmul -mlx::core::RandomBits -mlx::core::Reduce -mlx::core::Remainder -mlx::core::Reshape -mlx::core::Round -mlx::core::Scan -mlx::core::Scatter -mlx::core::Select -mlx::core::Sigmoid -mlx::core::Sign -mlx::core::Sin -mlx::core::Sinh -mlx::core::Slice -mlx::core::SliceUpdate -mlx::core::Softmax -mlx::core::Sort -mlx::core::Sqrt -mlx::core::Square -mlx::core::StopGradient -mlx::core::Subtract -mlx::core::Tan -mlx::core::Tanh -mlx::core::Transpose -mlx::core::Uniform -mlx::core::View +mlx::core::Imag +mlx::core::Inverse +mlx::core::Less +mlx::core::LessEqual +mlx::core::Load +mlx::core::Log +mlx::core::Log1p +mlx::core::LogAddExp +mlx::core::LogicalAnd +mlx::core::LogicalNot +mlx::core::LogicalOr +mlx::core::Matmul +mlx::core::Maximum +mlx::core::Minimum +mlx::core::Multiply +mlx::core::Negative +mlx::core::NotEqual +mlx::core::NumberOfElements +mlx::core::Pad +mlx::core::Partition +mlx::core::Power +mlx::core::QuantizedMatmul +mlx::core::RandomBits +mlx::core::Real +mlx::core::Reduce +mlx::core::Remainder +mlx::core::Reshape +mlx::core::Round +mlx::core::Scan +mlx::core::Scatter +mlx::core::Select +mlx::core::Sigmoid +mlx::core::Sign +mlx::core::Sin +mlx::core::Sinh +mlx::core::Slice +mlx::core::SliceUpdate +mlx::core::Softmax +mlx::core::Sort +mlx::core::Sqrt +mlx::core::Square +mlx::core::StopGradient +mlx::core::Subtract +mlx::core::Tan +mlx::core::Tanh +mlx::core::Transpose +mlx::core::Uniform +mlx::core::View
@@ -387,7 +389,7 @@ Public Member Functions
-

Implemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

+

Implemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

@@ -452,7 +454,7 @@ Public Member Functions
-

Implemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

+

Implemented in mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, and mlx::core::View.

diff --git a/docs/build/html/classmlx_1_1core_1_1_unary_primitive.png b/docs/build/html/classmlx_1_1core_1_1_unary_primitive.png index d3aa80740..fa6e8dc0f 100644 Binary files a/docs/build/html/classmlx_1_1core_1_1_unary_primitive.png and b/docs/build/html/classmlx_1_1core_1_1_unary_primitive.png differ diff --git a/docs/build/html/common_2unary_8h_source.html b/docs/build/html/common_2unary_8h_source.html index b933ced14..a6879dd7c 100644 --- a/docs/build/html/common_2unary_8h_source.html +++ b/docs/build/html/common_2unary_8h_source.html @@ -117,26 +117,26 @@ $(function(){ initResizable(false); });
24 }
25}
26
-
27template <typename T, typename Op>
-
28void unary_op(const T* a, T* out, Op op, size_t shape, size_t stride) {
+
27template <typename T, typename U = T, typename Op>
+
28void unary_op(const T* a, U* out, Op op, size_t shape, size_t stride) {
29 for (size_t i = 0; i < shape; i += 1) {
30 out[i] = op(*a);
31 a += stride;
32 }
33}
34
-
35template <typename T, typename Op>
+
35template <typename T, typename U = T, typename Op>
36void unary_op(const array& a, array& out, Op op) {
37 const T* a_ptr = a.data<T>();
38 if (a.flags().contiguous) {
39 set_unary_output_data(a, out);
-
40 T* dst = out.data<T>();
+
40 U* dst = out.data<U>();
41 for (size_t i = 0; i < a.data_size(); ++i) {
42 dst[i] = op(a_ptr[i]);
43 }
44 } else {
45 out.set_data(allocator::malloc_or_wait(out.nbytes()));
-
46 T* dst = out.data<T>();
+
46 U* dst = out.data<U>();
47 size_t shape = a.ndim() > 0 ? a.shape(-1) : 1;
48 size_t stride = a.ndim() > 0 ? a.strides(-1) : 1;
49 if (a.ndim() <= 1) {
diff --git a/docs/build/html/compiled_8h_source.html b/docs/build/html/compiled_8h_source.html index 0c7126b09..020b90fd1 100644 --- a/docs/build/html/compiled_8h_source.html +++ b/docs/build/html/compiled_8h_source.html @@ -176,7 +176,7 @@ $(function(){ initResizable(false); });
mlx::core::Broadcast
Definition primitives.h:528
mlx::core::Copy
Definition primitives.h:683
mlx::core::Primitive
Definition primitives.h:48
-
mlx::core::StopGradient
Definition primitives.h:1987
+
mlx::core::StopGradient
Definition primitives.h:2015
mlx::core::array
Definition array.h:20
mlx::core::array::ndim
size_t ndim() const
The number of dimensions of the array.
Definition array.h:94
mlx::core::array::item
T item()
Get the value from a scalar array.
Definition array.h:535
diff --git a/docs/build/html/cpp/ops.html b/docs/build/html/cpp/ops.html index 4c304b324..009ac5586 100644 --- a/docs/build/html/cpp/ops.html +++ b/docs/build/html/cpp/ops.html @@ -8,7 +8,7 @@ - Operations — MLX 0.18.1 documentation + Operations — MLX 0.19.0 documentation @@ -36,7 +36,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home +
@@ -325,6 +325,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -370,6 +371,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • @@ -1168,6 +1170,8 @@ document.write(`
  • roll()
  • roll()
  • roll()
    • +
  • real()
    • +
  • imag()
    • @@ -2967,6 +2971,16 @@ document.write(` array roll(const array &a, const std::vector<int> &shift, const std::vector<int> &axes, StreamOrDevice s = {})#
    +
    +
    +array real(const array &a, StreamOrDevice s = {})#
    +
    + +
    +
    +array imag(const array &a, StreamOrDevice s = {})#
    +
    + @@ -3320,6 +3334,8 @@ document.write(`
  • roll()
  • roll()
  • roll()
    • +
  • real()
    • +
  • imag()
    • diff --git a/docs/build/html/dev/custom_metal_kernels.html b/docs/build/html/dev/custom_metal_kernels.html index 19610ee45..37eaf2625 100644 --- a/docs/build/html/dev/custom_metal_kernels.html +++ b/docs/build/html/dev/custom_metal_kernels.html @@ -8,7 +8,7 @@ - Custom Metal Kernels — MLX 0.18.1 documentation + Custom Metal Kernels — MLX 0.19.0 documentation @@ -36,7 +36,7 @@ - + @@ -135,8 +135,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home +
    @@ -324,6 +324,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -369,6 +370,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • diff --git a/docs/build/html/dev/extensions.html b/docs/build/html/dev/extensions.html index 5763e3ecb..8c580ef55 100644 --- a/docs/build/html/dev/extensions.html +++ b/docs/build/html/dev/extensions.html @@ -8,7 +8,7 @@ - Custom Extensions in MLX — MLX 0.18.1 documentation + Custom Extensions in MLX — MLX 0.19.0 documentation @@ -36,7 +36,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home +
    @@ -325,6 +325,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -370,6 +371,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • diff --git a/docs/build/html/dev/metal_debugger.html b/docs/build/html/dev/metal_debugger.html index c68e205e3..ee6ca0493 100644 --- a/docs/build/html/dev/metal_debugger.html +++ b/docs/build/html/dev/metal_debugger.html @@ -8,7 +8,7 @@ - Metal Debugger — MLX 0.18.1 documentation + Metal Debugger — MLX 0.19.0 documentation @@ -36,7 +36,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home +
    @@ -325,6 +325,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -370,6 +371,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • diff --git a/docs/build/html/dir_70a37effa88bcbd6b791977fa1e64356.html b/docs/build/html/dir_70a37effa88bcbd6b791977fa1e64356.html index 78bb05f74..bcafd29fb 100644 --- a/docs/build/html/dir_70a37effa88bcbd6b791977fa1e64356.html +++ b/docs/build/html/dir_70a37effa88bcbd6b791977fa1e64356.html @@ -149,6 +149,8 @@ Files    scatter.h   + sdpa_vector.h +   softmax.h    sort.h diff --git a/docs/build/html/doxygen_crawl.html b/docs/build/html/doxygen_crawl.html index 3e85e314b..a17657db0 100644 --- a/docs/build/html/doxygen_crawl.html +++ b/docs/build/html/doxygen_crawl.html @@ -57,6 +57,7 @@ + @@ -185,6 +186,7 @@ + @@ -345,6 +347,8 @@ + + @@ -444,6 +448,8 @@ + + @@ -606,6 +612,8 @@ + + @@ -658,6 +666,8 @@ + + @@ -778,6 +788,8 @@ + + @@ -818,6 +830,8 @@ + + @@ -1791,10 +1805,10 @@ + - @@ -2428,6 +2442,16 @@ + + + + + + + + + + @@ -2655,6 +2679,16 @@ + + + + + + + + + + @@ -3970,6 +4004,7 @@ + @@ -4099,6 +4134,7 @@ + @@ -4202,7 +4238,6 @@ - @@ -4223,6 +4258,7 @@ + @@ -4312,9 +4348,9 @@ - - - + + + @@ -4624,6 +4660,7 @@ + @@ -4640,7 +4677,6 @@ - @@ -4859,7 +4895,6 @@ - @@ -5156,6 +5191,7 @@ + @@ -5560,6 +5596,7 @@ + @@ -5689,6 +5726,7 @@ + @@ -5880,6 +5918,8 @@ + + @@ -6090,6 +6130,8 @@ + + @@ -6332,6 +6374,8 @@ + + @@ -6637,6 +6681,8 @@ + + @@ -6680,6 +6726,8 @@ + + diff --git a/docs/build/html/examples/linear_regression.html b/docs/build/html/examples/linear_regression.html index f607b15ea..a53b1c7c9 100644 --- a/docs/build/html/examples/linear_regression.html +++ b/docs/build/html/examples/linear_regression.html @@ -8,7 +8,7 @@ - Linear Regression — MLX 0.18.1 documentation + Linear Regression — MLX 0.19.0 documentation @@ -36,7 +36,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home + @@ -325,6 +325,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -370,6 +371,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • diff --git a/docs/build/html/examples/llama-inference.html b/docs/build/html/examples/llama-inference.html index ee9554aa3..b7c837fb5 100644 --- a/docs/build/html/examples/llama-inference.html +++ b/docs/build/html/examples/llama-inference.html @@ -8,7 +8,7 @@ - LLM inference — MLX 0.18.1 documentation + LLM inference — MLX 0.19.0 documentation @@ -36,7 +36,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home + @@ -325,6 +325,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -370,6 +371,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • diff --git a/docs/build/html/examples/mlp.html b/docs/build/html/examples/mlp.html index 07cc5d702..8d9a1d75d 100644 --- a/docs/build/html/examples/mlp.html +++ b/docs/build/html/examples/mlp.html @@ -8,7 +8,7 @@ - Multi-Layer Perceptron — MLX 0.18.1 documentation + Multi-Layer Perceptron — MLX 0.19.0 documentation @@ -36,7 +36,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home + @@ -325,6 +325,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -370,6 +371,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • diff --git a/docs/build/html/files.html b/docs/build/html/files.html index 073a60747..3cefbb516 100644 --- a/docs/build/html/files.html +++ b/docs/build/html/files.html @@ -179,13 +179,14 @@ $(function(){ initResizable(false); });  scaled_dot_product_attention_params.h  scan.h  scatter.h - softmax.h - sort.h - ternary.h - ternary_ops.h - unary.h - unary_ops.h - utils.h + sdpa_vector.h + softmax.h + sort.h + ternary.h + ternary_ops.h + unary.h + unary_ops.h + utils.h  allocator.h  binary.h  copy.h diff --git a/docs/build/html/functions_e.html b/docs/build/html/functions_e.html index ff40389c0..813a4da96 100644 --- a/docs/build/html/functions_e.html +++ b/docs/build/html/functions_e.html @@ -99,8 +99,8 @@ $(function(){ initResizable(false); });
  • Erf() : mlx::core::Erf
  • ErfInv() : mlx::core::ErfInv
  • eval() : mlx::core::array
    • -
  • eval_cpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
    • -
  • eval_gpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
    • +
  • eval_cpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
    • +
  • eval_gpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
  • evaluated : mlx::core::array
  • Event() : mlx::core::Event
  • event() : mlx::core::array
    • diff --git a/docs/build/html/functions_func_e.html b/docs/build/html/functions_func_e.html index c594fdf3c..96cd83acc 100644 --- a/docs/build/html/functions_func_e.html +++ b/docs/build/html/functions_func_e.html @@ -96,8 +96,8 @@ $(function(){ initResizable(false); });
  • Erf() : mlx::core::Erf
  • ErfInv() : mlx::core::ErfInv
  • eval() : mlx::core::array
    • -
  • eval_cpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
    • -
  • eval_gpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
    • +
  • eval_cpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
    • +
  • eval_gpu() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Recv, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::AffineQuantize, mlx::core::fast::CustomKernel, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::UnaryPrimitive, mlx::core::Uniform, mlx::core::View
  • Event() : mlx::core::Event
  • event() : mlx::core::array
  • exec() : pocketfft::detail::cfftp< T0 >, pocketfft::detail::fftblue< T0 >, pocketfft::detail::pocketfft_c< T0 >, pocketfft::detail::pocketfft_r< T0 >, pocketfft::detail::rfftp< T0 >, pocketfft::detail::T_dcst23< T0 >, pocketfft::detail::T_dcst4< T0 >, pocketfft::detail::T_dct1< T0 >, pocketfft::detail::T_dst1< T0 >
    • diff --git a/docs/build/html/functions_func_i.html b/docs/build/html/functions_func_i.html index ee09ed40e..143886a5e 100644 --- a/docs/build/html/functions_func_i.html +++ b/docs/build/html/functions_func_i.html @@ -88,6 +88,7 @@ $(function(){ initResizable(false); });

    - i -

    diff --git a/docs/build/html/functions_func_p.html b/docs/build/html/functions_func_p.html index 80c5fe664..8ec35b141 100644 --- a/docs/build/html/functions_func_p.html +++ b/docs/build/html/functions_func_p.html @@ -99,7 +99,7 @@ $(function(){ initResizable(false); });
  • primitive() : mlx::core::array
  • primitive_id() : mlx::core::array
  • primitive_ptr() : mlx::core::array
    • -
  • print() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::PrintFormatter, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
    • +
  • print() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::PrintFormatter, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
  • prod() : pocketfft::detail::util
  • ptr() : mlx::core::allocator::Buffer
  • push() : pocketfft::detail::threading::concurrent_queue< T >
    • diff --git a/docs/build/html/functions_func_r.html b/docs/build/html/functions_func_r.html index 8b38a8b13..9284c3b09 100644 --- a/docs/build/html/functions_func_r.html +++ b/docs/build/html/functions_func_r.html @@ -94,6 +94,7 @@ $(function(){ initResizable(false); });
  • raw_ptr() : mlx::core::allocator::Buffer
  • read() : mlx::core::io::ParallelFileReader, mlx::core::io::Reader
  • ReadWriter() : ReadWriter< in_T, out_T, step, four_step_real >
    • +
  • Real() : mlx::core::Real
  • Recv() : mlx::core::distributed::Recv
  • Reduce() : mlx::core::Reduce
  • ReductionPlan() : mlx::core::ReductionPlan
    • diff --git a/docs/build/html/functions_func_v.html b/docs/build/html/functions_func_v.html index 039c7f84c..ee1802f11 100644 --- a/docs/build/html/functions_func_v.html +++ b/docs/build/html/functions_func_v.html @@ -91,8 +91,8 @@ $(function(){ initResizable(false); });
  • valid() : mlx::core::Event
  • value() : mlx::core::Event
  • View() : mlx::core::View
    • -
  • vjp() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::fast::LayerNorm, mlx::core::fast::RMSNorm, mlx::core::fast::RoPE, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose
    • -
  • vmap() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
    • +
  • vjp() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::fast::LayerNorm, mlx::core::fast::RMSNorm, mlx::core::fast::RoPE, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose
    • +
  • vmap() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
    • diff --git a/docs/build/html/functions_i.html b/docs/build/html/functions_i.html index dba895117..9d6403451 100644 --- a/docs/build/html/functions_i.html +++ b/docs/build/html/functions_i.html @@ -91,6 +91,7 @@ $(function(){ initResizable(false); });
  • id() : mlx::core::array
  • idil : MLXConvParams< NDIM >
  • idx_t : KernelMergeSort< T, U, ARG_SORT, BLOCK_THREADS, N_PER_THREAD, CompareOp >
    • +
  • Imag() : mlx::core::Imag
  • imag : complex64_t
  • in : ReadWriter< in_T, out_T, step, four_step_real >
  • in_strides : MLXConvParams< NDIM >
    • @@ -112,7 +113,7 @@ $(function(){ initResizable(false); });
  • iS : MLXConvParams< NDIM >
  • is_available() : mlx::core::array
  • is_donatable() : mlx::core::array
    • -
  • is_equivalent() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
    • +
  • is_equivalent() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::fast::ScaledDotProductAttention, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
  • is_open() : mlx::core::io::FileWriter, mlx::core::io::ParallelFileReader, mlx::core::io::Reader, mlx::core::io::Writer
  • is_ready() : pocketfft::detail::threading::latch
  • is_signaled() : mlx::core::Event
    • diff --git a/docs/build/html/functions_j.html b/docs/build/html/functions_j.html index 8ff8b0067..95cb4f61c 100644 --- a/docs/build/html/functions_j.html +++ b/docs/build/html/functions_j.html @@ -90,7 +90,7 @@ $(function(){ initResizable(false); });
  • jump_a : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • jump_b : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • jump_params : mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >
    • -
  • jvp() : mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose
    • +
  • jvp() : mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::Real, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose
    • diff --git a/docs/build/html/functions_o.html b/docs/build/html/functions_o.html index b6a0693a2..7d0b4afac 100644 --- a/docs/build/html/functions_o.html +++ b/docs/build/html/functions_o.html @@ -99,7 +99,7 @@ $(function(){ initResizable(false); });
  • operator T() : _MLX_BFloat16, complex64_t
  • operator Val() : mlx::core::Dtype
  • operator!= : mlx::core::array::ArrayIterator
    • -
  • operator()() : Abs, Add, ArcCos, ArcCosh, ArcSin, ArcSinh, ArcTan2, ArcTan, ArcTanh, BitwiseAnd, BitwiseOr, BitwiseXor, Ceil, Conjugate, Cos, Cosh, CumMax< U >, CumMin< U >, CumProd< U >, CumProd< bool >, CumSum< U >, Divide, DivMod, Equal, Erf, ErfInv, Exp, Expm1, Floor, FloorDivide, Greater, GreaterEqual, LeftShift, Less, LessEqual, LessThan< T >, Log10, Log1p, Log2, Log, LogAddExp, LogicalAnd, LogicalNot, LogicalOr, Maximum, Minimum, mlx::core::DefaultContiguousReduce< T, U, Op >, mlx::core::DefaultStridedReduce< T, U, Op >, mlx::core::detail::Abs, mlx::core::detail::Add, mlx::core::detail::ArcCos, mlx::core::detail::ArcCosh, mlx::core::detail::ArcSin, mlx::core::detail::ArcSinh, mlx::core::detail::ArcTan2, mlx::core::detail::ArcTan, mlx::core::detail::ArcTanh, mlx::core::detail::BitwiseAnd, mlx::core::detail::BitwiseOr, mlx::core::detail::BitwiseXor, mlx::core::detail::Ceil, mlx::core::detail::Conjugate, mlx::core::detail::Cos, mlx::core::detail::Cosh, mlx::core::detail::Divide, mlx::core::detail::Equal, mlx::core::detail::Erf, mlx::core::detail::ErfInv, mlx::core::detail::Exp, mlx::core::detail::Expm1, mlx::core::detail::Floor, mlx::core::detail::Greater, mlx::core::detail::GreaterEqual, mlx::core::detail::LeftShift, mlx::core::detail::Less, mlx::core::detail::LessEqual, mlx::core::detail::Log10, mlx::core::detail::Log1p, mlx::core::detail::Log2, mlx::core::detail::Log, mlx::core::detail::LogAddExp, mlx::core::detail::LogicalAnd, mlx::core::detail::LogicalNot, mlx::core::detail::LogicalOr, mlx::core::detail::Maximum, mlx::core::detail::Minimum, mlx::core::detail::Multiply, mlx::core::detail::NaNEqual, mlx::core::detail::Negative, mlx::core::detail::NotEqual, mlx::core::detail::Power, mlx::core::detail::Remainder, mlx::core::detail::RightShift, mlx::core::detail::Round, mlx::core::detail::Rsqrt, mlx::core::detail::Select, mlx::core::detail::Sigmoid, mlx::core::detail::Sign, mlx::core::detail::Sin, mlx::core::detail::Sinh, mlx::core::detail::Sqrt, mlx::core::detail::Square, mlx::core::detail::Subtract, mlx::core::detail::Tan, mlx::core::detail::Tanh, Multiply, NaNEqual, Negative, NotEqual, pocketfft::detail::ExecC2C, pocketfft::detail::ExecDcst, pocketfft::detail::ExecHartley, pocketfft::detail::ExecR2R, Power, Remainder, RightShift, Round, Rsqrt, Select, Sigmoid, Sign, Sin, Sinh, Sqrt, Square, Subtract, Tan, Tanh
    • +
  • operator()() : Abs, Add, ArcCos, ArcCosh, ArcSin, ArcSinh, ArcTan2, ArcTan, ArcTanh, BitwiseAnd, BitwiseOr, BitwiseXor, Ceil, Conjugate, Cos, Cosh, CumMax< U >, CumMin< U >, CumProd< U >, CumProd< bool >, CumSum< U >, Divide, DivMod, Equal, Erf, ErfInv, Exp, Expm1, Floor, FloorDivide, Greater, GreaterEqual, Imag, LeftShift, Less, LessEqual, LessThan< T >, Log10, Log1p, Log2, Log, LogAddExp, LogicalAnd, LogicalNot, LogicalOr, Maximum, Minimum, mlx::core::DefaultContiguousReduce< T, U, Op >, mlx::core::DefaultStridedReduce< T, U, Op >, mlx::core::detail::Abs, mlx::core::detail::Add, mlx::core::detail::ArcCos, mlx::core::detail::ArcCosh, mlx::core::detail::ArcSin, mlx::core::detail::ArcSinh, mlx::core::detail::ArcTan2, mlx::core::detail::ArcTan, mlx::core::detail::ArcTanh, mlx::core::detail::BitwiseAnd, mlx::core::detail::BitwiseOr, mlx::core::detail::BitwiseXor, mlx::core::detail::Ceil, mlx::core::detail::Conjugate, mlx::core::detail::Cos, mlx::core::detail::Cosh, mlx::core::detail::Divide, mlx::core::detail::Equal, mlx::core::detail::Erf, mlx::core::detail::ErfInv, mlx::core::detail::Exp, mlx::core::detail::Expm1, mlx::core::detail::Floor, mlx::core::detail::Greater, mlx::core::detail::GreaterEqual, mlx::core::detail::Imag, mlx::core::detail::LeftShift, mlx::core::detail::Less, mlx::core::detail::LessEqual, mlx::core::detail::Log10, mlx::core::detail::Log1p, mlx::core::detail::Log2, mlx::core::detail::Log, mlx::core::detail::LogAddExp, mlx::core::detail::LogicalAnd, mlx::core::detail::LogicalNot, mlx::core::detail::LogicalOr, mlx::core::detail::Maximum, mlx::core::detail::Minimum, mlx::core::detail::Multiply, mlx::core::detail::NaNEqual, mlx::core::detail::Negative, mlx::core::detail::NotEqual, mlx::core::detail::Power, mlx::core::detail::Real, mlx::core::detail::Remainder, mlx::core::detail::RightShift, mlx::core::detail::Round, mlx::core::detail::Rsqrt, mlx::core::detail::Select, mlx::core::detail::Sigmoid, mlx::core::detail::Sign, mlx::core::detail::Sin, mlx::core::detail::Sinh, mlx::core::detail::Sqrt, mlx::core::detail::Square, mlx::core::detail::Subtract, mlx::core::detail::Tan, mlx::core::detail::Tanh, Multiply, NaNEqual, Negative, NotEqual, pocketfft::detail::ExecC2C, pocketfft::detail::ExecDcst, pocketfft::detail::ExecHartley, pocketfft::detail::ExecR2R, Power, Real, Remainder, RightShift, Round, Rsqrt, Select, Sigmoid, Sign, Sin, Sinh, Sqrt, Square, Subtract, Tan, Tanh
  • operator*() : mlx::core::array::ArrayIterator, pocketfft::detail::cmplx< T >
  • operator*=() : pocketfft::detail::cmplx< T >
  • operator+() : mlx::core::array::ArrayIterator, pocketfft::detail::cmplx< T >
    • @@ -117,7 +117,7 @@ $(function(){ initResizable(false); });
  • out : ReadWriter< in_T, out_T, step, four_step_real >
  • out_of_bounds() : ReadWriter< in_T, out_T, step, four_step_real >
  • out_strides : MLXConvParams< NDIM >
    • -
  • output_shapes() : mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::Floor, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Round, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh
    • +
  • output_shapes() : mlx::core::Abs, mlx::core::Add, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::Floor, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Round, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Softmax, mlx::core::Sort, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh
  • outputs() : mlx::core::array
  • overwrite_descriptor() : mlx::core::array
    • diff --git a/docs/build/html/functions_p.html b/docs/build/html/functions_p.html index 569429d1b..bbacdcffd 100644 --- a/docs/build/html/functions_p.html +++ b/docs/build/html/functions_p.html @@ -102,7 +102,7 @@ $(function(){ initResizable(false); });
  • primitive() : mlx::core::array
  • primitive_id() : mlx::core::array
  • primitive_ptr() : mlx::core::array
    • -
  • print() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::PrintFormatter, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
    • +
  • print() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::Arange, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Load, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::PrintFormatter, mlx::core::QRF, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
  • Prod : mlx::core::distributed::AllReduce, mlx::core::Reduce, mlx::core::Scan, mlx::core::Scatter
  • prod() : pocketfft::detail::util
  • ptr() : mlx::core::allocator::Buffer
    • diff --git a/docs/build/html/functions_r.html b/docs/build/html/functions_r.html index 99e600d5d..b8d1b8eaa 100644 --- a/docs/build/html/functions_r.html +++ b/docs/build/html/functions_r.html @@ -100,6 +100,7 @@ $(function(){ initResizable(false); });
  • read_iw : mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >
  • read_n : mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >
  • ReadWriter() : ReadWriter< in_T, out_T, step, four_step_real >
    • +
  • Real() : mlx::core::Real
  • real : complex64_t
  • Recv() : mlx::core::distributed::Recv
  • Reduce() : mlx::core::Reduce
    • diff --git a/docs/build/html/functions_v.html b/docs/build/html/functions_v.html index dcbae8e9d..8bd2de211 100644 --- a/docs/build/html/functions_v.html +++ b/docs/build/html/functions_v.html @@ -96,8 +96,8 @@ $(function(){ initResizable(false); });
  • value_type : mlx::core::array::ArrayIterator, pocketfft::detail::threading::aligned_allocator< T >
  • vec_size : mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >, mlx::steel::ChannelHelper< n_channels_ >, mlx::steel::ChannelHelper< 1 >, mlx::steel::ChannelHelper< 2 >, mlx::steel::ChannelHelper< 3 >, mlx::steel::ChannelHelper< 4 >, mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >
  • View() : mlx::core::View
    • -
  • vjp() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::fast::LayerNorm, mlx::core::fast::RMSNorm, mlx::core::fast::RoPE, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose
    • -
  • vmap() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
    • +
  • vjp() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Convolution, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::fast::LayerNorm, mlx::core::fast::RMSNorm, mlx::core::fast::RoPE, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherMM, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose
    • +
  • vmap() : mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan2, mlx::core::ArcTan, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Cholesky, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::distributed::AllGather, mlx::core::distributed::AllReduce, mlx::core::distributed::Send, mlx::core::Divide, mlx::core::DivMod, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::fast::Custom, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::GatherQMM, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Imag, mlx::core::Inverse, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log1p, mlx::core::Log, mlx::core::LogAddExp, mlx::core::LogicalAnd, mlx::core::LogicalNot, mlx::core::LogicalOr, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::Primitive, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Real, mlx::core::Reduce, mlx::core::Remainder, mlx::core::Reshape, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Select, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Square, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::SVD, mlx::core::Tan, mlx::core::Tanh, mlx::core::Transpose, mlx::core::Uniform, mlx::core::View
    • diff --git a/docs/build/html/genindex.html b/docs/build/html/genindex.html index 952f3e74d..0555f3adf 100644 --- a/docs/build/html/genindex.html +++ b/docs/build/html/genindex.html @@ -7,7 +7,7 @@ - Index — MLX 0.18.1 documentation + Index — MLX 0.19.0 documentation @@ -35,7 +35,7 @@ - + @@ -135,8 +135,8 @@ - MLX 0.18.1 documentation - Home - + MLX 0.19.0 documentation - Home + @@ -324,6 +324,7 @@
  • mlx.core.greater_equal
  • mlx.core.hadamard_transform
  • mlx.core.identity
    • +
  • mlx.core.imag
  • mlx.core.inner
  • mlx.core.isfinite
  • mlx.core.isclose
    • @@ -369,6 +370,7 @@
  • mlx.core.quantize
  • mlx.core.quantized_matmul
  • mlx.core.radians
    • +
  • mlx.core.real
  • mlx.core.reciprocal
  • mlx.core.remainder
  • mlx.core.repeat
    • @@ -1480,6 +1482,10 @@ document.write(`
  • ifft2() (in module mlx.core.fft)
  • ifftn() (in module mlx.core.fft) +
    • +
  • imag (C++ function) +
    • +
  • imag() (in module mlx.core)
  • init() (in module mlx.core.distributed) @@ -1501,14 +1507,14 @@ document.write(`
  • irfftn() (in module mlx.core.fft)
    • + + - + -