Open
Description
When using the model.compile(backend="sendnn_decoder")
the backend shows the following warning:
You're using a deprecated backend. Please use sendnn in conjunction with warmup_mode
This is because we are not warming up using warmup_mode
with backend=“sendnn”
, it should be done as following:
model.forward = torch.compile(model.forward, backend = "sendnn", dynamic=False)
# warmup
with torch_sendnn.warmup_mode():
generate(model, ...)
It is currently being used in the following locations:
aiu-fms-testing-utils/scripts/inference.py
Line 467 in 6e7e360
aiu-fms-testing-utils/scripts/validation.py
Line 465 in 6e7e360
Metadata
Metadata
Assignees
Labels
No labels