-
Notifications
You must be signed in to change notification settings - Fork 12
All storage temporaries are now Quantity #377
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
| data = allocator(shape, dtype=dtype) | ||
| return Quantity(data, dims=dims, units=units, origin=origin, extent=extent) | ||
|
|
||
| def get_quantity_halo_spec( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should we cache this?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Maybe, but it's only called on init, and would need a bit of extra logic because you can't LRU cache on a list (not hashable).
| self.update_geopotential_height_on_c_grid( | ||
| self._zs, self._ut, self._vt, self._gz, self._ws3, dt2 | ||
| ) | ||
| # TODO (floriand): Due to DaCe VRAM pooling creating a memory |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this is solved, right? The fields are no longer used
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
self.cgrid_shallow_water_lagrangian_dynamics.ptc and self.cgrid_shallow_water_lagrangian_dynamics.delpc are used directly below. They used to be attached to this object instead of c_sw.
Purpose
This PR refactors all usage of storages to Quantity, other than in test code.
Code changes:
Checklist
Before submitting this PR, please make sure:
pace-util, HISTORY has been updated