Skip to content

Commit 428f2b5

Browse files
committed
fix
1 parent 6b9d985 commit 428f2b5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lightllm/distributed/communication_op.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@
5757
import deep_ep
5858
from deep_gemm.jit_kernels.utils import set_num_sms
5959

60-
deepep_sms = os.getenv("DEEPEP_SMS", deep_ep.Buffer.num_sms)
60+
deepep_sms = int(os.getenv("DEEPEP_SMS", deep_ep.Buffer.num_sms))
6161
device_sms = get_device_sm_count()
6262
deep_ep.Buffer.set_num_sms(deepep_sms)
6363
set_num_sms(device_sms - deepep_sms)

0 commit comments

Comments
 (0)