From f5f8ab2550380c5f223f5a9c34e209aac187c813 Mon Sep 17 00:00:00 2001 From: Austin Mroz Date: Tue, 4 Feb 2025 10:29:00 -0600 Subject: [PATCH] Support version calculation on AMD --- hy3dgen/texgen/custom_rasterizer/setup.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/hy3dgen/texgen/custom_rasterizer/setup.py b/hy3dgen/texgen/custom_rasterizer/setup.py index cab5d5a..b6767d6 100755 --- a/hy3dgen/texgen/custom_rasterizer/setup.py +++ b/hy3dgen/texgen/custom_rasterizer/setup.py @@ -3,9 +3,12 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension import torch torch_version = torch.__version__.split('+')[0].replace('.', '') -cuda_version = torch.version.cuda.replace('.', '') +if torch.version.cuda is not None: + cuda_version = "cuda"+torch.version.cuda.replace('.', '') +elif getattr(torch.version, 'hip', None) is not None: + cuda_version = "rocm" + torch.version.hip.replace('.','') -version = f"0.1.0+torch{torch_version}.cuda{cuda_version}" +version = f"0.1.0+torch{torch_version}.{cuda_version}" # build custom rasterizer # build with `python setup.py install` # nvcc is needed