423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-20 06:10:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

test fixes

Browse Source 
Summary: Some random seed changes. Skip multigpu tests when there's only one gpu. This is a better fix for what AI is doing in D80600882.

Reviewed By: MichaelRamamonjisoa

Differential Revision: D80625966

fbshipit-source-id: ac3952e7144125fd3a05ad6e4e6e5976ae10a8ef
This commit is contained in:
Jeremy Reizenstein
2025-08-27 06:55:50 -07:00
committed by Facebook GitHub Bot
parent 50f8efa1cb
commit dd068703d1
8 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -134,7 +134,7 @@ if os.getenv("PYTORCH3D_NO_NINJA", "0") == "1":
class BuildExtension(torch.utils.cpp_extension.BuildExtension):
def __init__(self, *args, **kwargs):
super().__init__(use_ninja=False, *args, **kwargs)
super().__init__(*args, use_ninja=False, **kwargs)
else:
BuildExtension = torch.utils.cpp_extension.BuildExtension