get_unchecked in tt and change repetition

Heiaha · Sep 2, 2023 · 51bff28 · 51bff28
1 parent 5fd70cd
commit 51bff28
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/src/board.rs b/src/board.rs
@@ -226,10 +226,9 @@ impl Board {
             self.history[self.ply].half_move_counter(),
         ) as usize;
 
-        self.history[..self.ply]
+        self.history[self.ply - lookback..self.ply]
             .iter()
             .rev()
-            .take(lookback)
             .skip(1)
             .step_by(2)
             .any(|entry| self.material_hash() == entry.material_hash())

diff --git a/src/piece.rs b/src/piece.rs
@@ -99,7 +99,7 @@ impl fmt::Display for Piece {
 }
 
 impl Piece {
-    pub const N_PIECES: usize = 13;
+    pub const N_PIECES: usize = 12;
     const PIECE_STR: &'static str = "PNBRQK  pnbrqk ";
 }
 

diff --git a/src/tt.rs b/src/tt.rs
@@ -99,13 +99,20 @@ impl TT {
 
     #[inline(always)]
     pub fn insert(&self, board: &Board, depth: Depth, value: Value, best_move: Move, flag: Bound) {
-        self.table[self.index(board)]
-            .write(board.hash(), TTEntry::new(value, best_move, depth, flag))
+        unsafe {
+            self.table
+                .get_unchecked(self.index(board))
+                .write(board.hash(), TTEntry::new(value, best_move, depth, flag))
+        }
     }
 
     #[inline(always)]
     pub fn probe(&self, board: &Board) -> Option<TTEntry> {
-        self.table[self.index(board)].read(board.hash())
+        unsafe {
+            self.table
+                .get_unchecked(self.index(board))
+                .read(board.hash())
+        }
     }
 
     pub fn clear(&mut self) {