Set 0 as the size of dummy tensors to reduce memory usage.#1

tdophung merged 0 commit intotdophung:teddy/jax-triton-permutationtdophung/TransformerEngine:teddy/jax-triton-permutationfrom

mingxu1067:mingh/set_dummy_with_zero_sizesmingxu1067/TransformerEngine:mingh/set_dummy_with_zero_sizesCopy head branch name to clipboard