-
Notifications
You must be signed in to change notification settings - Fork 45
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Updated CUDA Bindings for Passive Crossbar Inference Routines (#123)
Updated CUDA bindings for passive crossbar inference routines to avoid OOM errors (#120).
- Loading branch information
1 parent
913fce1
commit aec1e25
Showing
13 changed files
with
259 additions
and
333 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,3 +1,5 @@ | ||
at::Tensor solve_passive(at::Tensor conductance_matrix, at::Tensor V_WL, | ||
at::Tensor V_BL, float R_source, float R_line, | ||
bool det_readout_currents); | ||
at::Tensor V_BL, int ADC_resolution, | ||
float overflow_rate, int quant_method, | ||
float R_source, float R_line, | ||
bool det_readout_currents); |
Oops, something went wrong.