Optimize memory peak for _preprocess_state_vector
in LightningTensor
#9130
Job | Run time |
---|---|
6s | |
2m 55s | |
3m 8s | |
2m 53s | |
1m 55s | |
0s | |
10m 57s |
_preprocess_state_vector
in LightningTensor
#9130
Job | Run time |
---|---|
6s | |
2m 55s | |
3m 8s | |
2m 53s | |
1m 55s | |
0s | |
10m 57s |