Unverified Commit 924892fd authored by nzmora-nvidia's avatar nzmora-nvidia Committed by GitHub
Browse files

Update README.md (#27)



Fix the sample code so it compiles after the signature of `te.Linear` has changed.
Signed-off-by: default avatarnzmora-nvidia <96238833+nzmora-nvidia@users.noreply.github.com>
Signed-off-by: default avatarnzmora-nvidia <96238833+nzmora-nvidia@users.noreply.github.com>
parent 415caeb6
...@@ -48,7 +48,7 @@ Transformer Engine in action: ...@@ -48,7 +48,7 @@ Transformer Engine in action:
hidden_size = 2048 hidden_size = 2048
# Initialize model and inputs. # Initialize model and inputs.
model = te.Linear(in_features, out_features, use_bias=True) model = te.Linear(in_features, out_features, bias=True)
inp = torch.randn(hidden_size, in_features, device="cuda") inp = torch.randn(hidden_size, in_features, device="cuda")
# Create FP8 recipe. Note: All input args are optional. # Create FP8 recipe. Note: All input args are optional.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment