We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 96a3977 commit bbb8e00Copy full SHA for bbb8e00
slm/model_zoo/gpt-3/external_ops/token_dispatcher_utils/utils.h
@@ -124,5 +124,6 @@ __device__ __forceinline__ void vectorized_memcpy(const T* src,
124
PD_SWITCH_NUM_EXPERTS_IMPL(__num_expert, 8, __VA_ARGS__); \
125
PD_SWITCH_NUM_EXPERTS_IMPL(__num_expert, 16, __VA_ARGS__); \
126
PD_SWITCH_NUM_EXPERTS_IMPL(__num_expert, 32, __VA_ARGS__); \
127
+ PD_SWITCH_NUM_EXPERTS_IMPL(__num_expert, 64, __VA_ARGS__); \
128
PD_THROW("Unsupported expert number %d", int(__num_expert)); \
129
} while (0)
0 commit comments