Lines Matching defs:ExpandableSegment
356 struct ExpandableSegment { struct
357 ExpandableSegment( in ExpandableSegment() function
382 SegmentRange map(SegmentRange range) { in map()
424 SegmentRange unmap(SegmentRange range) { in unmap()
439 SegmentRange share(SegmentRange range, std::ostream& buf) { in share()
458 static std::unique_ptr<ExpandableSegment> fromShared( in fromShared()
517 char* ptr() const { in ptr()
522 size_t size() const { in size()
526 void addPeer(c10::DeviceIndex device) { in addPeer()
532 ~ExpandableSegment() { in ~ExpandableSegment()
540 void setAccess(c10::DeviceIndex device, size_t begin, size_t end) { in setAccess()
550 void mapAndSetAccess(size_t begin, size_t end) { in mapAndSetAccess()
565 void unmapHandles(size_t begin, size_t end) { in unmapHandles()
591 void trimHandles() { in trimHandles()
596 void forEachAllocatedRange(const std::function<void(size_t, size_t)>& fn) { in forEachAllocatedRange()
607 size_t numSegments(size_t size) { in numSegments()
610 size_t segmentLeft(char* p) { in segmentLeft()
614 size_t segmentRight(char* p) { in segmentRight()
618 SegmentRange rangeFromHandles(size_t begin, size_t end) { in rangeFromHandles()
622 c10::DeviceIndex device_;
623 std::optional<cudaStream_t> stream_;
624 CUdeviceptr ptr_{};
625 size_t segment_size_;
626 size_t max_handles_;
627 struct Handle {
631 struct ShareHeader {
636 std::vector<std::optional<Handle>> handles_;
639 std::vector<c10::DeviceIndex> peers_;
642 struct ExpandableSegment { struct
643 ExpandableSegment( in ExpandableSegment() function
651 SegmentRange map(SegmentRange range) { in map()
654 SegmentRange unmap(SegmentRange range) { in unmap()
657 SegmentRange share(SegmentRange range, std::ostream& ss) { in share()
660 static std::unique_ptr<ExpandableSegment> fromShared( in fromShared()
666 char* ptr() const { in ptr()
669 size_t size() const { in size()
672 void addPeer(c10::DeviceIndex device) {} in addPeer()