We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support NeMo's LLM MegatronGPTModel
MegatronGPTModel
Initial examine: Found 44 distinct operations, of which 33 (75%) are supported
TensorBase.div_
TensorBase.__setitem__
device
TensorBase.clone
TensorBase.view_as
_set_grad_enabled
TensorBase.is_cuda
TensorBase.is_complex
_VariableFunctionsClass.empty
_VariableFunctionsClass.max
_VariableFunctionsClass.baddbmm
cc @tfogal
The text was updated successfully, but these errors were encountered:
No branches or pull requests
馃殌 Feature
Support NeMo's LLM
MegatronGPTModel
Initial examine:
Found 44 distinct operations, of which 33 (75%) are supported
Motivation
Pitch
Work items
TensorBase.div_
#340TensorBase.__setitem__
#341device
of torch #327TensorBase.clone
#342TensorBase.view_as
#328_set_grad_enabled
of torch._C #332TensorBase.is_cuda
#329TensorBase.is_complex
#339_VariableFunctionsClass.empty
of torch #331_VariableFunctionsClass.max
#338_VariableFunctionsClass.baddbmm
of torch #326cc @tfogal
The text was updated successfully, but these errors were encountered: