fix no gpu/cpu

This commit is contained in:
Awni Hannun
2025-10-21 12:04:53 -07:00
parent 7b329a816e
commit 9b1ee2df33
2 changed files with 2 additions and 0 deletions

View File

@@ -130,6 +130,7 @@ NO_CPU(View)
namespace fast {
NO_CPU_MULTI(Quantize)
NO_CPU_MULTI(ConvertFP8)
} // namespace fast
namespace distributed {

View File

@@ -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