zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: metal formatting and pre-commit bump (#1038)

Browse Source 
* feat: metal formatting and pre-commit bump

* add guards

* update

* more guards

* more guards

* smakk fix

* Refactor instantiation of ternary types in ternary.metal

* fix scan.metal
This commit is contained in:
Nripesh Niketan
2024-04-30 18:18:09 +04:00
committed by GitHub
parent 8db7161c94
commit a30e7ed2da
45 changed files with 3822 additions and 3337 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/types/complex.h
View File

@@ -14,8 +14,8 @@ inline constexpr bool can_convert_to_complex128 =
!std::is_same_v<T, complex128_t> && std::is_convertible_v<T, double>;
struct complex128_t : public std::complex<double> {
complex128_t(double v, double u) : std::complex<double>(v, u){};
complex128_t(std::complex<double> v) : std::complex<double>(v){};
complex128_t(double v, double u) : std::complex<double>(v, u) {};
complex128_t(std::complex<double> v) : std::complex<double>(v) {};
template <
typename T,
@@ -32,8 +32,8 @@ inline constexpr bool can_convert_to_complex64 =
!std::is_same_v<T, complex64_t> && std::is_convertible_v<T, float>;
struct complex64_t : public std::complex<float> {
complex64_t(float v, float u) : std::complex<float>(v, u){};
complex64_t(std::complex<float> v) : std::complex<float>(v){};
complex64_t(float v, float u) : std::complex<float>(v, u) {};
complex64_t(std::complex<float> v) : std::complex<float>(v) {};
template <
typename T,