mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
@@ -23,7 +23,8 @@ namespace phi {
|
||||
|
||||
#ifdef PADDLE_CUDA_BF16
|
||||
template class MoeGemmRunner<
|
||||
__nv_bfloat16, cutlass::WintQuantTraits<__nv_bfloat16, cutlass::WintQuantMethod::kNone>>;
|
||||
__nv_bfloat16,
|
||||
cutlass::WintQuantTraits<__nv_bfloat16, cutlass::WintQuantMethod::kNone>>;
|
||||
#endif
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
@@ -24,7 +24,8 @@ namespace phi {
|
||||
#ifdef PADDLE_CUDA_BF16
|
||||
template class MoeGemmRunner<
|
||||
__nv_bfloat16,
|
||||
cutlass::WintQuantTraits<__nv_bfloat16, cutlass::WintQuantMethod::kWeightOnlyInt2>>;
|
||||
cutlass::WintQuantTraits<__nv_bfloat16,
|
||||
cutlass::WintQuantMethod::kWeightOnlyInt2>>;
|
||||
#endif
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
@@ -23,7 +23,8 @@ namespace phi {
|
||||
#ifdef PADDLE_CUDA_BF16
|
||||
template class MoeGemmRunner<
|
||||
__nv_bfloat16,
|
||||
cutlass::WintQuantTraits<__nv_bfloat16, cutlass::WintQuantMethod::kWeightOnlyInt4>>;
|
||||
cutlass::WintQuantTraits<__nv_bfloat16,
|
||||
cutlass::WintQuantMethod::kWeightOnlyInt4>>;
|
||||
#endif
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
@@ -24,7 +24,8 @@ namespace phi {
|
||||
#ifdef PADDLE_CUDA_BF16
|
||||
template class MoeGemmRunner<
|
||||
__nv_bfloat16,
|
||||
cutlass::WintQuantTraits<__nv_bfloat16, cutlass::WintQuantMethod::kWeightOnlyInt8>>;
|
||||
cutlass::WintQuantTraits<__nv_bfloat16,
|
||||
cutlass::WintQuantMethod::kWeightOnlyInt8>>;
|
||||
#endif
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
@@ -21,7 +21,8 @@
|
||||
|
||||
namespace phi {
|
||||
|
||||
template class MoeGemmRunner<half,
|
||||
cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kNone>>;
|
||||
template class MoeGemmRunner<
|
||||
half,
|
||||
cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kNone>>;
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
@@ -22,6 +22,7 @@
|
||||
namespace phi {
|
||||
|
||||
template class MoeGemmRunner<
|
||||
half, cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kWeightOnlyInt2>>;
|
||||
half,
|
||||
cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kWeightOnlyInt2>>;
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
@@ -22,6 +22,7 @@
|
||||
namespace phi {
|
||||
|
||||
template class MoeGemmRunner<
|
||||
half, cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kWeightOnlyInt4>>;
|
||||
half,
|
||||
cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kWeightOnlyInt4>>;
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
@@ -22,6 +22,7 @@
|
||||
namespace phi {
|
||||
|
||||
template class MoeGemmRunner<
|
||||
half, cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kWeightOnlyInt8>>;
|
||||
half,
|
||||
cutlass::WintQuantTraits<half, cutlass::WintQuantMethod::kWeightOnlyInt8>>;
|
||||
|
||||
} // namespace phi
|
||||
} // namespace phi
|
||||
|
||||
Reference in New Issue
Block a user