MCPcopy Create free account

hub / github.com/csguoh/OBR / functions

Functions645 in github.com/csguoh/OBR

⨍ Functions 645◇ Types & classes 88↳ Endpoints 2
↓ 1 callersFunctioneval_ppl
(args, model, tokenizer, device=torch.device("cuda:0"))
FlatQuant/flatquant/ppl_utils.py:86
↓ 1 callersFunctioneval_ppl_wikitext
(model, testenc, bs=1, device=None)
SpinQuant/ppl_utils.py:152
↓ 1 callersFunctioneval_ppl_wikitext
(model, testenc, bs=1, device=None)
QuaRot/ppl_utils.py:155
↓ 1 callersFunctioneval_ppl_wikitext
(model, testenc, bs=1, device=None)
FlatQuant/flatquant/ppl_utils.py:158
↓ 1 callersMethodfake_quant
(self, x)
FlatQuant/flatquant/quant_utils.py:77
↓ 1 callersMethodfasterquant
( self, blocksize=128, percdamp=0.01, groupsize=-1, actorder=False,
SpinQuant/eval_utils/gptq_utils.py:46
↓ 1 callersMethodfasterquant
( self, blocksize=128, percdamp=.01, groupsize=-1, actorder=False, static_groups=False )
FlatQuant/gptq_utils.py:52
↓ 1 callersMethodfasterquant_with_OBR
( self, blocksize=128, percdamp=.01, groupsize=-1, actorder=False, static_groups=False )
QuaRot/gptq_utils.py:130
↓ 1 callersMethodfind_params
(self, x)
SpinQuant/utils/quant_utils.py:391
↓ 1 callersMethodfind_params
(self, x)
QuaRot/quant_utils.py:286
↓ 1 callersMethodfind_params_per_token_groupwise
(self, x)
SpinQuant/utils/quant_utils.py:131
↓ 1 callersMethodfind_params_per_token_groupwise
(self, x)
QuaRot/quant_utils.py:121
↓ 1 callersMethodfind_params_weight_groupwise
(self, x)
SpinQuant/utils/quant_utils.py:333
↓ 1 callersFunctionfind_qlayers
(module, layers=[torch.nn.Linear, ], name='')
FlatQuant/flatquant/obr_utils.py:66
↓ 1 callersMethodforward
(self, x)
SpinQuant/eval_utils/modeling_llama.py:326
↓ 1 callersMethodforward
(self, x, R1)
SpinQuant/train_utils/modeling_llama_quant.py:328
↓ 1 callersMethodfree
(self)
SpinQuant/eval_utils/obr_utils.py:271
↓ 1 callersMethodfree
(self)
QuaRot/obr_utils.py:260
↓ 1 callersMethodfree
(self)
FlatQuant/gptq_utils.py:143
↓ 1 callersMethodfree
(self)
FlatQuant/flatquant/obr_utils.py:267
↓ 1 callersFunctionget_act_stats
(model, dataset)
FlatQuant/flatquant/flatness.py:21
↓ 1 callersFunctionget_c4
(nsamples, seed, seqlen, tokenizer)
SpinQuant/ppl_utils.py:43
↓ 1 callersFunctionget_c4
(nsamples, seed, seqlen, tokenizer)
QuaRot/ppl_utils.py:46
↓ 1 callersFunctionget_c4
(nsamples, seed, seqlen, tokenizer)
FlatQuant/flatquant/ppl_utils.py:47
↓ 1 callersFunctionget_c4_new
(nsamples, seed, seqlen, model, hf_token=None, eval_mode=False)
QuaRot/data_utils.py:30
↓ 1 callersFunctionget_c4_new
(nsamples, seed, seqlen, tokenizer, eval_mode=False)
FlatQuant/flatquant/data_utils.py:34
↓ 1 callersFunctionget_cuda_arch_flags
()
FlatQuant/setup.py:22
↓ 1 callersFunctionget_llama
(model_name, hf_token)
QuaRot/model_utils.py:110
↓ 1 callersFunctionget_llama
(model_name, hf_token)
FlatQuant/flatquant/model_utils.py:83
↓ 1 callersFunctionget_llama_31
(model_name, hf_token)
FlatQuant/flatquant/model_utils.py:97
↓ 1 callersFunctionget_loaders
(name, nsamples=128, seed=0, seqlen=2048, tokenizer=None)
SpinQuant/ppl_utils.py:72
↓ 1 callersFunctionget_loaders
(name, nsamples=128, seed=0, seqlen=2048, tokenizer=None)
QuaRot/ppl_utils.py:75
↓ 1 callersFunctionget_loaders
(name, nsamples=128, seed=0, seqlen=2048, tokenizer=None)
FlatQuant/flatquant/ppl_utils.py:78
↓ 1 callersFunctionget_local_rank
()
SpinQuant/utils/utils.py:158
↓ 1 callersFunctionget_logger
(logger_name: Optional[str])
SpinQuant/utils/utils.py:136
↓ 1 callersMethodget_matrix
(self, inv_t=False)
FlatQuant/flatquant/trans_utils.py:142
↓ 1 callersFunctionget_minq_maxq
(bits, sym)
SpinQuant/utils/quant_utils.py:21
↓ 1 callersFunctionget_model_type
(model)
QuaRot/model_utils.py:166
↓ 1 callersFunctionget_opt
(model_name)
QuaRot/model_utils.py:142
↓ 1 callersFunctionget_orthogonal_matrix
(size, mode, device=utils.DEV)
QuaRot/rotation_utils.py:114
↓ 1 callersFunctionget_pile
(nsamples, seed, seqlen, tokenizer)
FlatQuant/flatquant/data_utils.py:82
↓ 1 callersFunctionget_ptb_new
(nsamples, seed, seqlen, model, hf_token, eval_mode=False)
QuaRot/data_utils.py:70
↓ 1 callersFunctionget_ptb_new
(nsamples, seed, seqlen, tokenizer, eval_mode=False)
FlatQuant/flatquant/data_utils.py:62
↓ 1 callersFunctionget_qmin_qmax
(bits, sym)
FlatQuant/flatquant/quant_utils.py:10
↓ 1 callersFunctionget_qwen
(model_name)
QuaRot/model_utils.py:124
↓ 1 callersFunctionget_qwen2
(model_name, hf_token)
FlatQuant/flatquant/model_utils.py:111
↓ 1 callersFunctionget_random_orthg
(size)
FlatQuant/flatquant/function_utils.py:24
↓ 1 callersMethodget_scale_zero
(self, x)
FlatQuant/flatquant/quant_utils.py:85
↓ 1 callersMethodget_weight
(self, )
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:59
↓ 1 callersFunctionget_wikitext2
(nsamples, seed, seqlen, tokenizer)
SpinQuant/ppl_utils.py:21
↓ 1 callersFunctionget_wikitext2
(nsamples, seed, seqlen, model, hf_token, eval_mode=False)
QuaRot/data_utils.py:5
↓ 1 callersFunctionget_wikitext2
(nsamples, seed, seqlen, tokenizer)
QuaRot/ppl_utils.py:24
↓ 1 callersFunctionget_wikitext2
(nsamples, seed, seqlen, tokenizer, eval_mode=False)
FlatQuant/flatquant/data_utils.py:12
↓ 1 callersFunctionget_wikitext2
(nsamples, seed, seqlen, tokenizer)
FlatQuant/flatquant/ppl_utils.py:24
↓ 1 callersMethodgroup_texts
(self, examples)
SpinQuant/utils/data_utils.py:75
↓ 1 callersFunctionload_state_dict_from_HF
(local_dir)
SpinQuant/eval_utils/ptq.py:20
↓ 1 callersFunctionmain
()
QuaRot/main.py:19
↓ 1 callersFunctionmain
()
FlatQuant/main.py:19
↓ 1 callersFunctionmatrix_norm_one
(W)
SpinQuant/train_utils/optimizer.py:26
↓ 1 callersFunctionnorm
(v, dim: int = 1)
SpinQuant/train_utils/optimizer.py:21
↓ 1 callersFunctionobr_gptq_quantization
(W, quantizer, XTX, blocksize, W_mask, obr_alpha)
SpinQuant/eval_utils/obr_utils.py:32
↓ 1 callersFunctionobr_gptq_quantization
(W, quantizer, XTX, blocksize, W_mask, obr_alpha)
QuaRot/obr_utils.py:22
↓ 1 callersFunctionobr_gptq_quantization
(W, quantizer, XTX, blocksize, W_mask, obr_alpha)
FlatQuant/flatquant/obr_utils.py:20
↓ 1 callersFunctionobr_rtn_quantization
(W, quantizer)
SpinQuant/eval_utils/obr_utils.py:27
↓ 1 callersFunctionobr_rtn_quantization
(W, quantizer)
QuaRot/obr_utils.py:17
↓ 1 callersFunctionobr_rtn_quantization
(W, quantizer)
FlatQuant/flatquant/obr_utils.py:15
↓ 1 callersMethodoptimal_brain_restoration
(self, percdamp=.01,sparsity_ratio = 0.5, prune_n=0, prune_m = 0, obr_rtn=False,obr_alpha=0.5, blocksize=128,)
SpinQuant/eval_utils/obr_utils.py:197
↓ 1 callersMethodoptimal_brain_restoration
(self, percdamp=.01,sparsity_ratio = 0.5, prune_n=0, prune_m = 0, obr_rtn=False,obr_alpha=0.5, blocksize=128,)
QuaRot/obr_utils.py:186
↓ 1 callersMethodoptimal_brain_restoration
(self, percdamp=.01,sparsity_ratio = 0.5, prune_n=0, prune_m = 0, obr_rtn=False,obr_alpha=0.5, blocksize=128,)
FlatQuant/flatquant/obr_utils.py:192
↓ 1 callersFunctionparser_gen
()
SpinQuant/utils/process_args.py:47
↓ 1 callersFunctionpattern_match
(patterns, source_list)
SpinQuant/ppl_utils.py:204
↓ 1 callersFunctionpattern_match
(patterns, source_list)
QuaRot/ppl_utils.py:207
↓ 1 callersFunctionpattern_match
(patterns, source_list)
FlatQuant/flatquant/ppl_utils.py:211
↓ 1 callersFunctionprepare_model
(args, model)
SpinQuant/train_utils/main.py:17
↓ 1 callersFunctionpt_fsdp_state_dict
(model: torch.nn.Module)
SpinQuant/utils/utils.py:38
↓ 1 callersFunctionptq_model
(args, model, model_args=None)
SpinQuant/eval_utils/ptq.py:54
↓ 1 callersFunctionqr_retraction
(tan_vec)
SpinQuant/train_utils/optimizer.py:42
↓ 1 callersMethodquant_kcache
(self, q, k)
FlatQuant/flatquant/model_tools/llama31_utils.py:186
↓ 1 callersMethodquant_kcache
(self, q, k)
FlatQuant/flatquant/model_tools/qwen_utils.py:190
↓ 1 callersMethodquant_kcache
(self, q, k)
FlatQuant/flatquant/model_tools/llama_utils.py:186
↓ 1 callersMethodquant_vcache
(self, value_states)
FlatQuant/flatquant/model_tools/llama31_utils.py:179
↓ 1 callersMethodquant_vcache
(self, value_states)
FlatQuant/flatquant/model_tools/qwen_utils.py:183
↓ 1 callersMethodquant_vcache
(self, value_states)
FlatQuant/flatquant/model_tools/llama_utils.py:179
↓ 1 callersMethodquantize
(self, x)
QuaRot/quant_utils.py:351
↓ 1 callersMethodquantize_one_row
(self, x, row_idx)
SpinQuant/utils/quant_utils.py:482
↓ 1 callersMethodquantize_one_row
(self, x, row_idx)
FlatQuant/flatquant/quant_utils.py:223
↓ 1 callersFunctionrandom_orthogonal_matrix
Generate a random orthogonal matrix of the specified size. First, we generate a random matrix with entries from a standard distribution.
SpinQuant/eval_utils/rotation_utils.py:26
↓ 1 callersFunctionrandom_orthogonal_matrix
Generate a random orthogonal matrix of the specified size. First, we generate a random matrix with entries from a standard distribution.
QuaRot/rotation_utils.py:95
↓ 1 callersFunctionremove_unwanted_pytorch_nvcc_flags
()
FlatQuant/setup.py:9
↓ 1 callersMethodrep_matrix_only
(self, )
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:478
↓ 1 callersFunctionrotate_attention_inputs
(layer, R1)
SpinQuant/eval_utils/rotation_utils.py:63
↓ 1 callersFunctionrotate_attention_inputs
(layer, Q, model_type)
QuaRot/rotation_utils.py:133
↓ 1 callersFunctionrotate_attention_output
(layer, R1)
SpinQuant/eval_utils/rotation_utils.py:71
↓ 1 callersFunctionrotate_attention_output
(layer, Q, model_type)
QuaRot/rotation_utils.py:140
↓ 1 callersFunctionrotate_embeddings
(model, R1: torch.Tensor)
SpinQuant/eval_utils/rotation_utils.py:55
↓ 1 callersFunctionrotate_embeddings
(model, Q: torch.Tensor)
QuaRot/rotation_utils.py:124
↓ 1 callersFunctionrotate_head
(model, R1: torch.Tensor)
SpinQuant/eval_utils/rotation_utils.py:106
↓ 1 callersFunctionrotate_head
(model, Q: torch.Tensor)
QuaRot/rotation_utils.py:206
↓ 1 callersFunctionrotate_mlp_input
(layer, R1)
SpinQuant/eval_utils/rotation_utils.py:83
↓ 1 callersFunctionrotate_mlp_input
(layer, Q, model_type)
QuaRot/rotation_utils.py:156
← previousnext →201–300 of 645, ranked by callers