mirror of
https://github.com/denismhz/flake.git
synced 2025-11-09 16:16:23 +01:00
079f83ba45
- KoboldAI is no longer upstream maintained, so is now deprecated in nixified.ai - InvokeAI: v2.3.1.post2 -> v3.3.0post3 - textgen: init at v1.7 - treewide: update flake inputs including nixpkgs - treewide: add a bunch of new dependencies and upgrade old ones
31 lines
1 KiB
Diff
31 lines
1 KiB
Diff
From 20d90dfc2be8fedce229f47982db656862c9dc32 Mon Sep 17 00:00:00 2001
|
|
From: Paul Mulders <justinkb@gmail.com>
|
|
Date: Thu, 11 May 2023 00:43:51 +0200
|
|
Subject: [PATCH] setup.py: fix ROCm build (#7573)
|
|
|
|
---
|
|
setup.py | 8 +++++++-
|
|
1 file changed, 7 insertions(+), 1 deletion(-)
|
|
|
|
diff --git a/setup.py b/setup.py
|
|
index c523ba073c5..732b5c0e1b7 100644
|
|
--- a/setup.py
|
|
+++ b/setup.py
|
|
@@ -328,9 +328,15 @@ def get_extensions():
|
|
image_src = (
|
|
glob.glob(os.path.join(image_path, "*.cpp"))
|
|
+ glob.glob(os.path.join(image_path, "cpu", "*.cpp"))
|
|
- + glob.glob(os.path.join(image_path, "cuda", "*.cpp"))
|
|
)
|
|
|
|
+ if is_rocm_pytorch:
|
|
+ image_src += glob.glob(os.path.join(image_path, "hip", "*.cpp"))
|
|
+ # we need to exclude this in favor of the hipified source
|
|
+ image_src.remove(os.path.join(image_path, "image.cpp"))
|
|
+ else:
|
|
+ image_src += glob.glob(os.path.join(image_path, "cuda", "*.cpp"))
|
|
+
|
|
if use_png or use_jpeg:
|
|
ext_modules.append(
|
|
extension(
|