Skip to content

Commit 51812f1

Browse files
committed
change arch_condition header dependency
1 parent 521fe01 commit 51812f1

File tree

2 files changed

+2
-102
lines changed

2 files changed

+2
-102
lines changed

csrc/fused_moe/moeTopKFuncs.cuh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,14 +23,14 @@
2323

2424
#include <cub/cub.cuh>
2525

26-
#include "tensorrt_llm/kernels/archCondition.h"
26+
#include "flashinfer/arch_condition.h"
2727

2828
namespace tensorrt_llm::kernels {
2929

3030
namespace reduce_topk {
3131
namespace cg = cooperative_groups;
3232
static constexpr int kWARP_SIZE = 32;
33-
static constexpr bool kTLLM_GEN_HAS_FAST_REDUX = tensorrt_llm::kernels::arch::is_major_v<10>;
33+
static constexpr bool kTLLM_GEN_HAS_FAST_REDUX = flashinfer::arch::is_major_v<10>;
3434

3535
template <typename T_>
3636
struct TopKRedType {

csrc/nv_internal/tensorrt_llm/kernels/archCondition.h

Lines changed: 0 additions & 100 deletions
This file was deleted.

0 commit comments

Comments
 (0)