fix: remove cuda/functional from MoE topk kernels (not available on corex)
This commit is contained in:
@@ -24,7 +24,6 @@ limitations under the License.
|
||||
#include <cub/util_type.cuh>
|
||||
|
||||
#if !defined(USE_DCU) && !defined(USE_MACA)
|
||||
#include <cuda/functional>
|
||||
#endif
|
||||
|
||||
#include "kernels/cuda/device_utils.cuh"
|
||||
|
||||
@@ -24,7 +24,6 @@ limitations under the License.
|
||||
#include <cub/util_type.cuh>
|
||||
|
||||
#if !defined(USE_DCU) && !defined(USE_MACA)
|
||||
#include <cuda/functional>
|
||||
#endif
|
||||
|
||||
#include "kernels/cuda/device_utils.cuh"
|
||||
|
||||
Reference in New Issue
Block a user