github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-21 21:22:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
flybird11111 2025-04-09 17:29:08 +08:00
parent e92a692c97
commit 0e900ac5cd
3 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/build_on_pr.yml vendored
View File

@ -161,7 +161,7 @@ jobs:
--ignore tests/test_infer_ops \ --ignore tests/test_infer_ops \
--ignore tests/test_legacy \ --ignore tests/test_legacy \
--ignore tests/test_smoothquant \ --ignore tests/test_smoothquant \
tests/test_fp8/ tests/
env: env:
LD_LIBRARY_PATH: /github/home/.tensornvme/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 LD_LIBRARY_PATH: /github/home/.tensornvme/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
LLAMA_PATH: /data/scratch/llama-tiny LLAMA_PATH: /data/scratch/llama-tiny

2
tests/test_device/test_init_logical_pg.py
View File

@ -1,4 +1,3 @@
import pytest
import torch import torch
import torch.distributed as dist import torch.distributed as dist
from torch.distributed import ReduceOp from torch.distributed import ReduceOp
@ -27,7 +26,6 @@ def check_layer(rank, world_size, port):
assert tensor.equal(tensor_to_check) assert tensor.equal(tensor_to_check)
@pytest.mark.skip("tested in corresponding sharderformer")
@pytest.mark.dist @pytest.mark.dist
@rerun_if_address_is_in_use() @rerun_if_address_is_in_use()
def test_logical_pg(): def test_logical_pg():

2
tests/test_fp8/test_fp8_allgather.py
View File

@ -13,7 +13,7 @@ from colossalai.testing import parameterize, rerun_if_address_is_in_use, spawn,
"shape", "shape",
[(3, 7, 16)], [(3, 7, 16)],
) )
@clear_cache_before_run()
@parameterize("dtype", [torch.bfloat16, torch.float16]) @parameterize("dtype", [torch.bfloat16, torch.float16])
@parameterize("fp8_format", ["e4m3", "e5m2"]) @parameterize("fp8_format", ["e4m3", "e5m2"])
@parameterize("async_op", [True, False]) @parameterize("async_op", [True, False])