fix handcode_opt bert [pr] (#8509)

* fix handcode_opt bert [pr]

* too slow
This commit is contained in:
chenyu
2025-01-05 19:14:12 -05:00
committed by GitHub
parent 9229867fec
commit b6be407bc6

View File

@@ -50,7 +50,7 @@ def get_sched_bert():
# ignore grad norm and loss scaler for now
loss.backward()
targets += [x for x in optim.schedule_step()]
sched = Tensor.schedule(targets)
sched = Tensor.schedule(*targets)
print(f"schedule length {len(sched)}")
return sched