mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
fix no gpu/cpu
This commit is contained in:
@@ -130,6 +130,7 @@ NO_CPU(View)
|
||||
|
||||
namespace fast {
|
||||
NO_CPU_MULTI(Quantize)
|
||||
NO_CPU_MULTI(ConvertFP8)
|
||||
} // namespace fast
|
||||
|
||||
namespace distributed {
|
||||
|
||||
@@ -154,6 +154,7 @@ NO_GPU_USE_FALLBACK(RMSNorm)
|
||||
NO_GPU_MULTI(RMSNormVJP)
|
||||
NO_GPU_USE_FALLBACK(RoPE)
|
||||
NO_GPU(ScaledDotProductAttention)
|
||||
NO_GPU_MULTI(ConvertFP8)
|
||||
NO_GPU_MULTI(Quantize)
|
||||
NO_GPU_MULTI(CustomKernel)
|
||||
} // namespace fast
|
||||
|
||||
Reference in New Issue
Block a user