Index

_

_ask_restrictions() (llm_kernel_tuner.tuning_strategies.BaseTuningStrategy method)

_extract_and_sanitize_kernel() (llm_kernel_tuner.tuning_strategies.BaseTuningStrategy method) _run_tests() (llm_kernel_tuner.tuning_strategies.BaseTuningStrategy method)

A

add_test() (llm_kernel_tuner.LLMKernelTransformer method)

AutonomousTuningStrategy (class in llm_kernel_tuner.tuning_strategies)

B

BaseTestingStrategy (class in llm_kernel_tuner.testing_strategies) BaseTuningStrategy (class in llm_kernel_tuner.tuning_strategies)

best_params (llm_kernel_tuner.tuning_state.State attribute) best_tune_params (llm_kernel_tuner.PerformanceStep attribute)

C

copy() (llm_kernel_tuner.TunableKernel method) create_graph() (llm_kernel_tuner.tuning_strategies.BaseTuningStrategy method)

create_retry_wrapper() (in module llm_kernel_tuner.retry) curr_tune_params (llm_kernel_tuner.tuning_state.State attribute)

D

DEFAULT (llm_kernel_tuner.structured_output.StructuredOutputType attribute)

E

ExplicitTuningStrategy (class in llm_kernel_tuner.tuning_strategies)

G

generate_overview() (llm_kernel_tuner.PerformanceTracker method) get_arg_position() (llm_kernel_tuner.TunableKernel method)

get_structured_llm() (in module llm_kernel_tuner.structured_output) get_test_from_code() (llm_kernel_tuner.testing_strategies.BaseTestingStrategy method) get_total_improvement() (llm_kernel_tuner.PerformanceTracker method)

H

has_improvements() (llm_kernel_tuner.PerformanceTracker method)

HYBRID_JSONIFY (llm_kernel_tuner.structured_output.StructuredOutputType attribute)

I

improvement_percentage (llm_kernel_tuner.PerformanceStep attribute)

J

JSON_SCHEMA (llm_kernel_tuner.structured_output.StructuredOutputType attribute)

K

kernel (llm_kernel_tuner.tuning_state.State attribute)

kernel_code (llm_kernel_tuner.PerformanceStep attribute) KernelTest (class in llm_kernel_tuner)

L

llm (llm_kernel_tuner.tuning_state.State attribute) llm_kernel_tuner.prompts.autonomous_tuning_strategy.agent_prompt (built-in variable) llm_kernel_tuner.prompts.autonomous_tuning_strategy.breakdown_step_prompt (built-in variable) llm_kernel_tuner.prompts.autonomous_tuning_strategy.fix_params_prompt (built-in variable) llm_kernel_tuner.prompts.autonomous_tuning_strategy.initial_planning_prompt (built-in variable) llm_kernel_tuner.prompts.autonomous_tuning_strategy.replan_prompt (built-in variable) llm_kernel_tuner.prompts.autonomous_tuning_strategy.validate_step_prompt (built-in variable) llm_kernel_tuner.prompts.explicit_strategy_prompts.step_evaluation_prompt (built-in variable) llm_kernel_tuner.prompts.explicit_strategy_prompts.system_prompt (built-in variable), [1] llm_kernel_tuner.prompts.get_restrictions_prompt (built-in variable) llm_kernel_tuner.prompts.one_prompt_strategy_prompts.fix_params_prompt (built-in variable) llm_kernel_tuner.prompts.one_prompt_strategy_prompts.system_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.compile_error_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.default_error_prompt (built-in variable), [1] llm_kernel_tuner.prompts.retry_prompts.default_tester_error_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.failed_tests_prompt (built-in variable)

llm_kernel_tuner.prompts.retry_prompts.invalid_output_variables_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.invalid_problem_size_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.invalid_restrictions_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.invalid_test_generated_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.no_code_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.timeout_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.wrong_stucture_prompt (built-in variable) llm_kernel_tuner.prompts.retry_prompts.wrong_tune_params_prompt (built-in variable) llm_kernel_tuner.prompts.tester_prompts.system_prompt (built-in variable) llm_kernel_tuner.prompts.tester_prompts.test_prompt (built-in variable) llm_kernel_tuner.prompts.transformer_prompts.extract_output_var_prompt (built-in variable) llm_kernel_tuner.prompts.transformer_prompts.kernel_description_prompt (built-in variable) llm_kernel_tuner.prompts.transformer_prompts.problem_size_prompt (built-in variable) llm_kernel_tuner.structured_output module LLMKernelTransformer (class in llm_kernel_tuner)

M

make_kernel_tunable() (llm_kernel_tuner.LLMKernelTransformer method) messages (llm_kernel_tuner.tuning_state.State attribute)

module llm_kernel_tuner.structured_output

N

new_execution_time (llm_kernel_tuner.PerformanceStep attribute)

O

old_execution_time (llm_kernel_tuner.PerformanceStep attribute)

P

performance_tracker (llm_kernel_tuner.tuning_state.State attribute)

PerformanceStep (class in llm_kernel_tuner) PerformanceTracker (class in llm_kernel_tuner)

R

record_step() (llm_kernel_tuner.PerformanceTracker method)

restrictions (llm_kernel_tuner.PerformanceStep attribute) RetryPolicy (class in llm_kernel_tuner.retry)

S

SEPARATE_REQUEST (llm_kernel_tuner.structured_output.StructuredOutputType attribute) set_baseline_time() (llm_kernel_tuner.PerformanceTracker method)

State (class in llm_kernel_tuner.tuning_state) step_description (llm_kernel_tuner.PerformanceStep attribute) StructuredOutputType (class in llm_kernel_tuner.structured_output)

T

test() (llm_kernel_tuner.TunableKernel method) tests (llm_kernel_tuner.tuning_state.State attribute) timestamp (llm_kernel_tuner.PerformanceStep attribute) TOOL_CALLING (llm_kernel_tuner.structured_output.StructuredOutputType attribute)

tunable_parameters (llm_kernel_tuner.PerformanceStep attribute) TunableKernel (class in llm_kernel_tuner) tune() (llm_kernel_tuner.TunableKernel method) TuningStep (class in llm_kernel_tuner.tuning_strategies)