[TensorExpr] fix bug in IRSimplifier when multiplying by 0 (#36287)
Summary:
In the IR Simplifier we were not treating multiply by zero specially, which meant some constant expressions were stored in formats that were not constant.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36287
Differential Revision: D20937497
Pulled By: nickgg
fbshipit-source-id: 528e430313ea048524d7a4a0256eef4a0297438b