Skip to content

Commit

Permalink
Merge pull request #377 from chaoming0625/master
Browse files Browse the repository at this point in the history
[bug] fix analysis jit bug
  • Loading branch information
chaoming0625 authored May 24, 2023
2 parents ec485fc + 60b3e89 commit 75e4368
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion brainpy/_src/analysis/lowdim/lowdim_analyzer.py
Original file line number Diff line number Diff line change
Expand Up @@ -436,7 +436,7 @@ def F_fy(self):
f = partial(f, **(self.pars_update + self.fixed_vars))
f = utils.f_without_jaxarray_return(f)
f = utils.remove_return_shape(f)
self.analyzed_results[C.F_fy] = bm.jit(f, device=self.jit_device)
self.analyzed_results[C.F_fy] = jax.jit(f, device=self.jit_device)
return self.analyzed_results[C.F_fy]

@property
Expand Down

0 comments on commit 75e4368

Please sign in to comment.