await: directly check for context.Canceled errors:

[dfuentes]

• Use a direct equality check to tell if a suboutine was ended due to a context.Canceled error. The old check using errors.Is() would falsely catch library errors that themselves wrap context.Canceled. In those cases, we should propogate the error instead of exiting cleanly.