Continuing from discussion 1-2 weeks ago on the extra allocations in evaluation of spherical harmonics observed in EQM. The reason is that PooledSparseProductLayer did not release layer properly.
Major updates from this PR:
Update PooledSparseProductLayer, added the release_input field
Broadcast release_input in PolyLuxLayer for tuple inputs
Minor fix:
printing format in LinearLayer's test
I also ran the test of EQM locally and it works fine.
cortner
commented
8 months ago
Continuing from discussion 1-2 weeks ago on the extra allocations in evaluation of spherical harmonics observed in EQM. The reason is that
PooledSparseProductLayerdid not release layer properly.Major updates from this PR:
PooledSparseProductLayer, added therelease_inputfieldrelease_inputinPolyLuxLayerfor tuple inputsMinor fix:
LinearLayer's testI also ran the test of EQM locally and it works fine.