Functions — github.com/Jellyfish042/RWKV_Othello

CodeHub MCPcopy Create free account

hub / github.com/Jellyfish042/RWKV_Othello / functions

Functions154 in github.com/Jellyfish042/RWKV_Othello

⨍ Functions 154◇ Types & classes 15

↓ 71 callersFunctionmatmul

(a, b, mx: Optional[torch.Tensor]=None, rx: Optional[torch.Tensor]=None, my: Optional[torch.Tensor]=None, ry:

rwkv_extended.py:127

↓ 8 callersFunctionformat_score

Format an integer score with proper sign and padding. Examples: 5 -> "+05" -3 -> "-03" 21 -> "+21" -15 -> "-1

formatter.py:21

↓ 7 callersMethod_get_flips

(self, row, col)

↓ 7 callersFunctionformat_board

Format the board to a human-readable string. 0 = empty (·) 1 = black (●) 2 = white (○)

↓ 6 callersMethodplay_from_start

(self, moves: str)

↓ 6 callersMethodupdate_value

(self, variable, delta)

↓ 5 callersMethodcleanup

↓ 5 callersMethodget_legal_moves

↓ 5 callersMethodplay

(self, moves: str)

↓ 5 callersMethodsend_command

(self, command, allow_restart=True)

↓ 4 callersFunctionformat_color

formatter.py:82

↓ 4 callersMethodis_game_over

↓ 4 callersMethodmake_ai_move

↓ 3 callersMethodcreate_spinner

(self, parent, variable, decrease_cmd, increase_cmd)

↓ 3 callersFunctioncuda_mm8_one

(N: int, M: int, x, w, mx, rx, my, ry)

rwkv_extended.py:76

↓ 3 callersFunctioncuda_mm8_seq

(B: int, N: int, M: int, x, w, mx, rx, my, ry)

rwkv_extended.py:64

↓ 3 callersMethoddraw_piece

(self, canvas, color, is_indicator=False)

↓ 3 callersFunctionformat_args

(max_width, max_depth)

formatter.py:78

↓ 3 callersFunctionformat_possible_moves

(moves_list, sep=" ")

formatter.py:43

↓ 3 callersFunctionformat_stack

formatter.py:86

↓ 3 callersMethodget_best_move

(self, moves: str)

↓ 3 callersMethodget_moves

↓ 3 callersMethodreset

↓ 3 callersMethodupdate_board

↓ 2 callersMethod_calculate_value_range

(self, all_values)

↓ 2 callersMethod_start_engine

↓ 2 callersMethoddraw_evaluation_graph

(self, event=None)

↓ 2 callersFunctionformat_possible_moves_no_sort

(moves_list, sep=" ")

formatter.py:64

↓ 2 callersMethodforward_one

(self, idx:int, state:List[torch.Tensor])

rwkv_extended.py:287

↓ 2 callersMethodget_avg_tokens

rwkv_engine.py:45

↓ 2 callersMethodget_moves

(self, moves: str)

↓ 2 callersFunctionmm8_one

(x, w, mx, rx, my, ry)

rwkv_extended.py:108

↓ 2 callersFunctionmm8_seq

(x, w, mx, rx, my, ry)

rwkv_extended.py:101

↓ 2 callersMethodon_player_change

↓ 2 callersMethodplay_one

(self, player1_color, opening=None)

↓ 2 callersMethodshow_game_result

↓ 2 callersFunctionsort_positions

(positions, ascending=True)

formatter.py:127

↓ 2 callersFunctiontorch_mm8_one

(x, w, mx, rx, my, ry)

rwkv_extended.py:96

↓ 2 callersFunctiontorch_mm8_seq

(x, w, mx, rx, my, ry)

rwkv_extended.py:92

↓ 2 callersMethodv5_2_after

(self, t_decay, out, s, x, xxx, g, lx_w, lx_b, ow, omx, orx, omy, ory)

rwkv_extended.py:1234

↓ 1 callersMethodRUN_RWKV_5

(self, B, T, C, H, state, r, k, v, w, u)

rwkv_extended.py:830

↓ 1 callersMethodRUN_RWKV_6

(self, B, T, C, H, state, r, k, v, w, u)

rwkv_extended.py:833

↓ 1 callersFunctionRWKV7_OP

(state, r, w, k, v, a, b)

rwkv_extended.py:218

↓ 1 callersFunctionRWKV_x070_CMix_one

(x, x_prev, x_k, K_, V_)

rwkv_extended.py:452

↓ 1 callersFunctionRWKV_x070_CMix_seq

(x, x_prev, x_k, K_, V_)

rwkv_extended.py:459

↓ 1 callersFunctionRWKV_x070_TMix_one

(layer_id: int, H:int, N:int, x, x_prev, v_first, state, x_r, x_w, x_k, x_v, x_a, x_g, w0, w1, w2, a0, a1, a2,

rwkv_extended.py:354

↓ 1 callersFunctionRWKV_x070_TMix_seq

(layer_id: int, H:int, N:int, x, x_prev, v_first, state, x_r, x_w, x_k, x_v, x_a, x_g, w0, w1, w2, a0, a1, a2,

rwkv_extended.py:388

↓ 1 callersMethod__init__

(self, model, strategy, verbose = True, convert_and_save_and_exit = None)

rwkv_extended.py:468

↓ 1 callersMethod_convert_position

↓ 1 callersMethod_stream_save_result

generate_data.py:26

↓ 1 callersMethodcallback

rwkv_engine.py:36

↓ 1 callersMethodclear

↓ 1 callersMethodcreate_board

↓ 1 callersMethodcreate_board_area

↓ 1 callersMethodcreate_evaluation_graph

↓ 1 callersMethodcreate_info_area

↓ 1 callersMethodcreate_model_settings

↓ 1 callersMethodcreate_player_settings

↓ 1 callersMethodcreate_reasoning_area

↓ 1 callersMethodcreate_status_display

↓ 1 callersFunctioncuda_wkv

(T: int, C: int, w, u, k, v, aa, bb, pp)

rwkv_extended.py:52

↓ 1 callersMethoddraw_graph_grid

(self, width, height)

↓ 1 callersMethoddraw_graph_lines

(self, width, height)

↓ 1 callersMethoddraw_valid_move

↓ 1 callersMethodevaluate

↓ 1 callersFunctionformat_input_moves

formatter.py:147

↓ 1 callersMethodforward_seq

(self, idx:List[int], state:List[torch.Tensor], full_output:bool=False)

rwkv_extended.py:320

↓ 1 callersMethodgen_one_sample

(self, input_moves, max_width, max_depth)

generate_data.py:59

↓ 1 callersMethodgenerate_samples_parallel

(self, inputs: List[tuple[str, int, int]], timeout: int = 120, batch_size: int = 1000)

generate_data.py:40

↓ 1 callersMethodget_best_move

(self, input_moves, max_token_count=1000000)

rwkv_engine.py:51

↓ 1 callersMethodget_best_move

Get the best move given the previous moves. Use Alpha-Beta pruning algorithm to search the game tree.

alphabeta_engine.py:12

↓ 1 callersMethodget_board_format

↓ 1 callersMethodget_move_from_player

(self, player, prev_moves)

↓ 1 callersFunctionget_position_value

formatter.py:130

↓ 1 callersMethodhighlight_cell

↓ 1 callersMethodmake_move

(self, row, col)

↓ 1 callersFunctionmatmul_float

(a, b, output_dtype: Optional[torch.dtype]=None)

rwkv_extended.py:144

↓ 1 callersFunctionmm8

(x: torch.Tensor, w: torch.Tensor, mx: torch.Tensor, rx: torch.Tensor, my: torch.Tensor, ry: torch.Tensor)

rwkv_extended.py:122

↓ 1 callersMethodnew_game

↓ 1 callersFunctionparallel_generate

( engine_path, engine_level, engine_threads, game_path, sample_per_game, output_file,

generate_data.py:117

↓ 1 callersMethodplay

(self, moves: str)

↓ 1 callersFunctionposition_to_value

formatter.py:50

↓ 1 callersFunctionpower_pairs

(limit, max_x=10, max_y=10)

generate_data.py:152

↓ 1 callersFunctionread_all_txt_files

generate_data.py:65

↓ 1 callersMethodreset

↓ 1 callersMethodrestart

Restart the engine by cleaning up existing process and starting a new one.

↓ 1 callersFunctionsample_game_states

(games, sample_per_game, length_weight=0, min_prob=0.01)

generate_data.py:88

↓ 1 callersMethodset_state_by_moves

↓ 1 callersMethodsetup_ui

↓ 1 callersMethodthink

(self, board_state, current_player, move_count, callback, search_config)

rwkv_for_ui.py:36

↓ 1 callersMethodunhighlight_cell

↓ 1 callersMethodupdate_current_indicator

↓ 1 callersMethodv5_2_before

(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, k_mix, v_mix, r_mix, g_mix, t_decay, t_first, kw, vw, rw, gw, ow, kmx

rwkv_extended.py:1218

↓ 1 callersMethodv6_0_before

(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, x_maa, w_maa, k_maa, v_maa, r_maa, g_maa, tm_w1, tm_w2, td_w1, td_w2,

rwkv_extended.py:1259

FunctionPYBIND11_MODULE

cuda/wrapper.cpp:125

FunctionTORCH_LIBRARY

cuda/wrapper.cpp:134

FunctionTORCH_LIBRARY

cuda/rwkv7_op.cpp:22

generate_data.py:50

(self, path, print_output=False, max_width=1, max_depth=1, rwkv_version=None)

rwkv_engine.py:14

(self, model, strategy)

rwkv_extended.py:224

next →1–100 of 154, ranked by callers