NVIDIA / NVIDIA/apex

Recent commits raise this error for multi_tensor_lamb_stage1_cuda

Open
#890 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
9k
Forks
1.5k
Avg merge
2d 4h
Merged PRs (30d)
3

Description

TypeError: multi_tensor_lamb_stage1_cuda(): incompatible function arguments. The following argument types are supported:
    1. (arg0: int, arg1: at::Tensor, arg2: List[List[at::Tensor]], arg3: at::Tensor, arg4: int, arg5: float, arg6: float, arg7: float, arg8: at::Tensor, arg9: float) -> None

I see this error on current HEAD: c3fad1ad120b23055f6630da0b029c8b626db78f
I don't see this error on commit from May 14 : 3bae8c83494184673f01f3867fa051518e930895

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by comparing the reported commits c3fad1ad120b23055f6630da0b029c8b626db78f and 3bae8c83494184673f01f3867fa051518e930895, then reproduce the multi_tensor_lamb_stage1_cuda TypeError on the current HEAD. Trace the function binding and recent changes to identify the regression; done means the error no longer occurs for the reported call.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.