Use a non-trivial benchmark for kernel_tests/benchmark_test.py
Otherwise XLA constant folds the computation resulting in allocator_maximum_num_bytes_* getting a value of 0. At least that's what I think is happening, I have not dug beyond confirming that the failure goes away with this change. PiperOrigin-RevId: 222435753
Loading
Please sign in to comment