Unverified Commit f478aa77 authored by Wang, Yi's avatar Wang, Yi Committed by GitHub
Browse files

hotfix: ipex fails since cuda moe kernel is not supported (#2532)


Signed-off-by: default avatarWang, Yi A <yi.a.wang@intel.com>
parent abd24dd3
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
from typing import List, Optional, Tuple from typing import List, Optional, Tuple
from moe_kernels.fused_moe import grouped_topk
import torch import torch
import torch.distributed import torch.distributed
from text_generation_server.layers import ( from text_generation_server.layers import (
...@@ -41,6 +40,9 @@ from torch import nn ...@@ -41,6 +40,9 @@ from torch import nn
from transformers.activations import ACT2FN from transformers.activations import ACT2FN
from transformers.configuration_utils import PretrainedConfig from transformers.configuration_utils import PretrainedConfig
if SYSTEM != "ipex":
from moe_kernels.fused_moe import grouped_topk
if SYSTEM == "rocm": if SYSTEM == "rocm":
try: try:
from vllm import _custom_C from vllm import _custom_C
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment