You may want to enter multiple context managers in one with-statement.
However, with rng_context and fp8_context is equivalent to with fp8_context, because rng_context doesn't rewrite __and__ method.
The correct way is with rng_context, fp8_context: .
You may want to enter multiple context managers in one with-statement. However,
with rng_context and fp8_contextis equivalent towith fp8_context, becauserng_contextdoesn't rewrite__and__method. The correct way iswith rng_context, fp8_context:.Ref: https://docs.python.org/3/reference/compound_stmts.html#the-with-statement