Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/deepspeedai/DeepSpeedExamples
/ forward
Method
forward
Megatron-LM/fp16/fp16util.py:97–99 ·
view source on GitHub ↗
(self, *inputs)
Source
from the content-addressed store, hash-verified
95
self.network = convert_network(network, dtype=torch.half)
96
97
def
forward(self, *inputs):
98
inputs = tuple(t.half()
for
t in inputs)
99
return
self.network(*inputs)
100
101
102
def
backwards_debug_hook(grad):
Callers
nothing calls this directly
Calls
1
half
Method · 0.80
Tested by
no test coverage detected