From 1f92f7cc6103effba9871d408ee45cb04c444463 Mon Sep 17 00:00:00 2001
From: Atream <boxin-zhang@qq.com>
Date: Thu, 8 Aug 2024 16:02:42 +0800
Subject: [PATCH] [fix] linux and windows can all find CPUInfer in current
 Directory

---
 ktransformers/operators/experts.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/ktransformers/operators/experts.py b/ktransformers/operators/experts.py
index bc44342..35821d7 100644
--- a/ktransformers/operators/experts.py
+++ b/ktransformers/operators/experts.py
@@ -19,8 +19,9 @@ import torch
 import sys, os
 from ktransformers.operators.base_operator import BaseInjectedModule
 
-#sys.path.append(os.path.dirname(__file__) + "/../ktransformers_ext/build/")
-sys.path.append(os.path.dirname(__file__) + "\\..\\ktransformers_ext\\build\\Release")
+sys.path.append(os.path.join(os.path.dirname(__file__), "..", "ktransformers_ext", "build"))
+sys.path.append(os.path.join(os.path.dirname(__file__), "..", "ktransformers_ext", "build", "Release"))
+sys.path.append(os.path.join(os.path.dirname(__file__), "..", "ktransformers_ext", "build", "Debug"))
 import cpuinfer_ext
 from cpuinfer_ext.moe import MOEConfig, MOE
 import ctypes