From aff5e26fa2d03d7185021c16b4d47af655660d0c Mon Sep 17 00:00:00 2001 From: flybird11111 <1829166702@qq.com> Date: Thu, 6 Jun 2024 03:20:02 +0000 Subject: [PATCH 1/2] fix --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index b105c03b717c..d2cfb13a4dfd 100644 --- a/setup.py +++ b/setup.py @@ -144,6 +144,7 @@ def get_version() -> str: package_data={ "colossalai": [ "kernel/extensions/csrc/**/*", + "kernel/extensions/pybind/**/*", ] }, ) From 7d2ce1ec782ae0be53841e47627d6f53681adc93 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 6 Jun 2024 03:21:28 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- extensions/cuda_extension.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions/cuda_extension.py b/extensions/cuda_extension.py index edfbf6f24e06..da15bcd57a26 100644 --- a/extensions/cuda_extension.py +++ b/extensions/cuda_extension.py @@ -30,7 +30,7 @@ def is_available(self) -> bool: # torch.cuda.is_available requires a device to exist, allow building with cuda extension on build nodes without a device # but where cuda is actually available. - cuda_available = torch.cuda.is_available() or bool(os.environ.get('FORCE_CUDA', 0)) + cuda_available = torch.cuda.is_available() or bool(os.environ.get("FORCE_CUDA", 0)) except: cuda_available = False return cuda_available