Skip to content
This repository has been archived by the owner on Aug 7, 2024. It is now read-only.

Commit

Permalink
try forcing nightly
Browse files Browse the repository at this point in the history
  • Loading branch information
drisspg committed Jun 22, 2024
1 parent c8b6050 commit 7983b78
Show file tree
Hide file tree
Showing 4 changed files with 3 additions and 5 deletions.
1 change: 1 addition & 0 deletions .github/workflows/python-app.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ jobs:
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip3 install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
pip install -e .
pip install -e .'[dev]'
pip install -e .'[test]'
Expand Down
4 changes: 1 addition & 3 deletions float8_experimental/float8_dynamic_linear.py
Original file line number Diff line number Diff line change
Expand Up @@ -174,9 +174,7 @@ def cast_to_float8_e4m3fn(
if activation_scale is not None
else tensor_to_scale(inpt_tensor, e4m3_dtype, reduce_amax)
)
return Float8Tensor.to_float8(
inpt_tensor, scale, e4m3_dtype, mm_config=mm_config
)
return Float8Tensor.to_float8(inpt_tensor, scale, e4m3_dtype, mm_config=mm_config)


def cast_to_float8_e5m2_bw(
Expand Down
1 change: 0 additions & 1 deletion float8_experimental/float8_python_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
to simplify the product code.
"""


from typing import Optional

import float8_experimental.float8_aten_api # noqa
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ classifiers = [
]

dependencies = [
"torch >= 2.2",
"torch >= 2.3",
]

[project.optional-dependencies]
Expand Down

0 comments on commit 7983b78

Please sign in to comment.