-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Summary: The main pytorch wheels on PyPI support CUDA 10.2. Here we make pytorch3d's wheels do the same, instead of being cpu only. This should ultimately make life easier in colab. Also a little script to count builds, which can be useful for nightly jobs. Reviewed By: gkioxari Differential Revision: D22924321 fbshipit-source-id: d6cea9bfbab49bcb0080f65608066c553ea8bb4d
- Loading branch information
1 parent
07d7e12
commit 4872a2c
Showing
4 changed files
with
38 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,29 @@ | ||
#!/usr/bin/env python3 | ||
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved. | ||
|
||
""" | ||
Print the number of nightly builds | ||
""" | ||
|
||
from collections import Counter | ||
|
||
import yaml | ||
|
||
|
||
conf = yaml.safe_load(open("config.yml")) | ||
jobs = conf["workflows"]["build_and_test"]["jobs"] | ||
|
||
|
||
def jobtype(job): | ||
if isinstance(job, str): | ||
return job | ||
if len(job) == 1: | ||
[name] = job.keys() | ||
return name | ||
return "MULTIPLE PARTS" | ||
|
||
|
||
for i, j in Counter(map(jobtype, jobs)).items(): | ||
print(i, j) | ||
print() | ||
print(len(jobs)) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters