MCPcopy Create free account
hub / github.com/ml-explore/mlx / ~PyCheckpointedFun

Method ~PyCheckpointedFun

python/src/transforms.cpp:593–597  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

591 public:
592 PyCheckpointedFun(nb::callable fun) : fun_(std::move(fun)) {}
593 ~PyCheckpointedFun() {
594 nb::gil_scoped_acquire gil;
595
596 fun_.reset();
597 }
598
599 struct InnerFunction {
600 nb::object fun_;

Callers

nothing calls this directly

Calls 1

resetMethod · 0.45

Tested by

no test coverage detected