Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add deepspeed #19021

Closed
wants to merge 6 commits into from
Closed

Add deepspeed #19021

Changes from 5 commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
9ff33de
Add deepspeed
weiji14 May 20, 2022
90ddae5
Set minimum Python version to 3.6
weiji14 May 20, 2022
c0edd95
Use hjson-py as runtime dependency
weiji14 May 28, 2022
bf9e1e0
Add ninja as host dependency
weiji14 May 28, 2022
aa848d9
[skip ci] Revert adding ninja as host dependency
weiji14 May 28, 2022
a8b489c
Merge branch 'conda-forge:main' into deepspeed
weiji14 Nov 2, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
50 changes: 50 additions & 0 deletions recipes/deepspeed/meta.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
{% set name = "deepspeed" %}
{% set version = "0.6.4" %}

package:
name: {{ name|lower }}
version: {{ version }}

source:
url: https://pypi.io/packages/source/{{ name[0] }}/{{ name }}/deepspeed-{{ version }}.tar.gz
sha256: ad2de46d6609798fa3525ef8eb0e1ddfb66e06983b0af967b462b1511f21cc9c

build:
script: {{ PYTHON }} -m pip install . -vv
noarch: python
number: 0

requirements:
host:
- pip
- python >=3.6
- pytorch
run:
- hjson-py
- ninja
Copy link
Member Author

@weiji14 weiji14 May 28, 2022 •
edited

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This ninja isn't the PyPI one from https://pypi.org/project/ninja/. See issue at conda-forge/ninja-feedstock#26. Need to have a python-ninja package or similar on conda-forge first. Edit: PR at #19098

- numpy
- packaging
- psutil
- py-cpuinfo
- python >=3.6
- pytorch
- tqdm

test:
imports:
- deepspeed
commands:
- pip check
requires:
- pip

about:
home: http://deepspeed.ai
summary: DeepSpeed library
dev_url: https://github.com/microsoft/DeepSpeed
license: MIT
license_file: LICENSE

extra:
recipe-maintainers:
- weiji14