Unverified Commit 93cf7fc5 authored by c1lovez1's avatar c1lovez1 Committed by GitHub
Browse files

Unify variable naming: replace is_in_free_group with is_not_in_free_group (#4698)

parent 2a206b22
...@@ -190,7 +190,7 @@ class TokenToKVPoolAllocator: ...@@ -190,7 +190,7 @@ class TokenToKVPoolAllocator:
self.free_slots = torch.arange( self.free_slots = torch.arange(
1, self.size + 1, dtype=torch.int64, device=self.device 1, self.size + 1, dtype=torch.int64, device=self.device
) )
self.is_in_free_group = False self.is_not_in_free_group = True
self.free_group = [] self.free_group = []
......
...@@ -279,5 +279,5 @@ class PagedTokenToKVPoolAllocator: ...@@ -279,5 +279,5 @@ class PagedTokenToKVPoolAllocator:
self.free_pages = torch.arange( self.free_pages = torch.arange(
1, self.num_pages + 1, dtype=torch.int64, device=self.device 1, self.num_pages + 1, dtype=torch.int64, device=self.device
) )
self.is_in_free_group = False self.is_not_in_free_group = True
self.free_group = [] self.free_group = []
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment