Index

A

append_config_parser() (in module lorien.configs) AutoSchedulerJob (class in lorien.dialect.tvm_dial.auto_scheduler_dial.job) AutoSchedulerJobConfigs (class in lorien.dialect.tvm_dial.auto_scheduler_dial.job) AutoSchedulerRecords (class in lorien.dialect.tvm_dial.auto_scheduler_dial.result) AutoSchedulerTuneResult (class in lorien.dialect.tvm_dial.auto_scheduler_dial.result) AutoSchedulerWorkload (class in lorien.dialect.tvm_dial.auto_scheduler_dial.workload)

AutoTVMJob (class in lorien.dialect.tvm_dial.autotvm_dial.job) AutoTVMJobConfigs (class in lorien.dialect.tvm_dial.autotvm_dial.job) AutoTVMRecords (class in lorien.dialect.tvm_dial.autotvm_dial.result) AutoTVMTuneResult (class in lorien.dialect.tvm_dial.autotvm_dial.result) AutoTVMWorkload (class in lorien.dialect.tvm_dial.autotvm_dial.workload) AWSBatchJobManager (class in lorien.tune.job_manager)

C

callback() (lorien.dialect.tvm_dial.auto_scheduler_dial.job.RecordToMetadata method) callback_metadata() (in module lorien.dialect.tvm_dial.autotvm_dial.job) check_table() (in module lorien.database.table) check_tvm_build_config() (lorien.dialect.tvm_dial.job.TVMJobConfigs method) check_tvm_version() (in module lorien.dialect.tvm_dial.util) ClientThread (class in lorien.tune.rpc.launch) commit() (lorien.tune.result.Records method) (lorien.tune.result.TuneResult method) commit_tuning_log() (lorien.tune.result.TuneResult method) convert_to_db_dict() (in module lorien.database.util) convert_to_db_list() (in module lorien.database.util) convert_to_dict() (in module lorien.database.util)

convert_to_list() (in module lorien.database.util) create_autotvm_tuner() (in module lorien.dialect.tvm_dial.autotvm_dial.job) create_config_parser() (in module lorien.configs) create_job_configs() (lorien.dialect.tvm_dial.auto_scheduler_dial.job.AutoSchedulerJob static method) (lorien.dialect.tvm_dial.autotvm_dial.job.AutoTVMJob static method) (lorien.dialect.tvm_dial.job.TVMJob static method) (lorien.tune.job.Job static method) create_layout_transform_task() (in module lorien.dialect.tvm_dial.autotvm_dial.extract_from_record) create_records_by_workloads() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerTuneResult static method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMTuneResult static method) (lorien.tune.result.TuneResult static method) create_table() (in module lorien.database.table)

D

decode() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords static method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords static method) (lorien.dialect.tvm_dial.result.TVMRecords static method) (lorien.tune.result.Records static method) deep_tuple_to_list() (in module lorien.util) delete_s3_file() (in module lorien.util) delete_table() (in module lorien.database.table)

desc() (lorien.tune.job_manager.AWSBatchJobManager method) (lorien.tune.job_manager.JobManagerBase method) (lorien.tune.job_manager.LocalJobManager method) (lorien.tune.job_manager.RPCJobManager method) disable_stream_handler() (in module lorien.logger) download_s3_file() (in module lorien.util) dump_to_yaml() (in module lorien.util)

E

enable_log_file() (in module lorien.logger) encode() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords static method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords static method) (lorien.dialect.tvm_dial.result.TVMRecords static method) (lorien.tune.result.Records static method)

extract_feature() (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMTuneResult static method) extract_feature_from_file() (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMTuneResult static method) extract_from_models() (in module lorien.dialect.tvm_dial.auto_scheduler_dial.extract) (in module lorien.dialect.tvm_dial.autotvm_dial.extract_from_model) extract_from_records() (in module lorien.dialect.tvm_dial.autotvm_dial.extract_from_record)

F

fetch_results() (lorien.tune.rpc.client.RPCClient method) from_task() (lorien.dialect.tvm_dial.auto_scheduler_dial.workload.AutoSchedulerWorkload class method) (lorien.dialect.tvm_dial.autotvm_dial.workload.AutoTVMWorkload class method) (lorien.workload.Workload class method)

G

gen() (in module lorien.generate) gen_features() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerTuneResult static method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMTuneResult static method) (lorien.tune.result.TuneResult static method) gen_record_item() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords static method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords static method) (lorien.dialect.tvm_dial.result.TVMRecords static method) (lorien.tune.result.Records static method) gen_target_id_keys() (in module lorien.dialect.tvm_dial.util) gen_task_item() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords method) (lorien.dialect.tvm_dial.result.TVMRecords method) (lorien.tune.result.Records method) get_canonical_tvm_target_str() (in module lorien.dialect.tvm_dial.util)

get_framework_build_config() (lorien.dialect.tvm_dial.result.TVMRecords method) (lorien.tune.result.Records method) get_job_configs_str() (lorien.tune.rpc.server.RPCService method) get_log_file_name() (lorien.workload.Workload method) get_logger() (in module lorien.logger) get_manager_configs() (in module lorien.tune.master) get_metadata() (lorien.tune.job.Job method) get_target() (in module lorien.tune.master) get_time_str() (in module lorien.util) get_tvm_build_config() (in module lorien.dialect.tvm_dial.util) get_tvm_container_value_str() (in module lorien.dialect.tvm_dial.util) get_workload_key() (lorien.dialect.tvm_dial.auto_scheduler_dial.workload.AutoSchedulerWorkload method) (lorien.dialect.tvm_dial.autotvm_dial.workload.AutoTVMWorkload method) (lorien.workload.Workload method)

H

hash_sha2() (lorien.workload.Workload method)

I

infer_task_layout() (in module lorien.dialect.tvm_dial.autotvm_dial.util) init() (lorien.tune.rpc.server.RPCService method) init_server() (lorien.tune.rpc.client.RPCClient method) init_worker() (lorien.tune.rpc.client.RPCClient method) is_cover() (in module lorien.dialect.tvm_dial.util)

is_dialect_enabled() (in module lorien.util) is_init() (lorien.tune.rpc.server.RPCService method) is_server_init() (lorien.tune.rpc.client.RPCClient method) is_target_compatible() (lorien.dialect.tvm_dial.job.TVMJob method) (lorien.tune.job.Job method)

J

Job (class in lorien.tune.job) JobConfigs (class in lorien.tune.job) JobEventLogger (class in lorien.tune.job)

JobManagerBase (class in lorien.tune.job_manager) JobState (class in lorien.tune.job) join() (lorien.tune.rpc.launch.ClientThread method)

L

launch_client() (in module lorien.tune.rpc.launch) launch_server() (in module lorien.tune.rpc.launch) list_tables() (in module lorien.database.table) load_from_yaml() (in module lorien.util) localize() (lorien.dialect.tvm_dial.auto_scheduler_dial.job.AutoSchedulerJobConfigs method) (lorien.dialect.tvm_dial.autotvm_dial.job.AutoTVMJobConfigs method) (lorien.dialect.tvm_dial.job.TVMJobConfigs method) (lorien.tune.job.JobConfigs method) LocalJobManager (class in lorien.tune.job_manager) lorien.configs module lorien.database.table module lorien.database.util module lorien.dialect.tvm_dial.auto_scheduler_dial.extract module lorien.dialect.tvm_dial.auto_scheduler_dial.job module lorien.dialect.tvm_dial.auto_scheduler_dial.result module lorien.dialect.tvm_dial.auto_scheduler_dial.workload module lorien.dialect.tvm_dial.autotvm_dial.extract_from_model module lorien.dialect.tvm_dial.autotvm_dial.extract_from_record module lorien.dialect.tvm_dial.autotvm_dial.job module lorien.dialect.tvm_dial.autotvm_dial.result module lorien.dialect.tvm_dial.autotvm_dial.util module

lorien.dialect.tvm_dial.autotvm_dial.workload module lorien.dialect.tvm_dial.frontend_parser module lorien.dialect.tvm_dial.job module lorien.dialect.tvm_dial.result module lorien.dialect.tvm_dial.util module lorien.generate module lorien.logger module lorien.main module lorien.tune.job module lorien.tune.job_manager module lorien.tune.master module lorien.tune.result module lorien.tune.rpc.client module lorien.tune.rpc.launch module lorien.tune.rpc.server module lorien.util module lorien.workload module

M

Main (class in lorien.main) make_config_parser() (in module lorien.configs) match_socket_port() (lorien.tune.rpc.server.RPCService method) module lorien.configs lorien.database.table lorien.database.util lorien.dialect.tvm_dial.auto_scheduler_dial.extract lorien.dialect.tvm_dial.auto_scheduler_dial.job lorien.dialect.tvm_dial.auto_scheduler_dial.result lorien.dialect.tvm_dial.auto_scheduler_dial.workload lorien.dialect.tvm_dial.autotvm_dial.extract_from_model lorien.dialect.tvm_dial.autotvm_dial.extract_from_record lorien.dialect.tvm_dial.autotvm_dial.job lorien.dialect.tvm_dial.autotvm_dial.result lorien.dialect.tvm_dial.autotvm_dial.util lorien.dialect.tvm_dial.autotvm_dial.workload lorien.dialect.tvm_dial.frontend_parser lorien.dialect.tvm_dial.job lorien.dialect.tvm_dial.result lorien.dialect.tvm_dial.util lorien.generate lorien.logger lorien.main lorien.tune.job lorien.tune.job_manager lorien.tune.master lorien.tune.result lorien.tune.rpc.client lorien.tune.rpc.launch lorien.tune.rpc.server lorien.util lorien.workload

mutate() (lorien.dialect.tvm_dial.auto_scheduler_dial.workload.AutoSchedulerWorkload method) (lorien.dialect.tvm_dial.autotvm_dial.workload.AutoTVMWorkload method) (lorien.workload.Workload method)

N

need_root() (in module lorien.tune.rpc.server) num_jobs() (lorien.tune.job_manager.JobManagerBase method)

num_workers() (lorien.tune.rpc.client.RPCClient method) (lorien.tune.rpc.server.RPCService method)

O

on_connect() (lorien.tune.rpc.server.RPCService method)

on_disconnect() (lorien.tune.rpc.server.RPCService method)

P

parse_manager_info() (in module lorien.tune.master) parse_model_config() (in module lorien.dialect.tvm_dial.frontend_parser) peak() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords method) (lorien.dialect.tvm_dial.result.TVMRecords method) (lorien.tune.result.Records method) pop() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords method) (lorien.dialect.tvm_dial.result.TVMRecords method) (lorien.tune.result.Records method)

prune_table() (in module lorien.dialect.tvm_dial.util) prune_table_item() (in module lorien.dialect.tvm_dial.util) push() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords method) (lorien.dialect.tvm_dial.result.TVMRecords method) (lorien.tune.result.Records method)

Q

query() (lorien.tune.result.Records method)

R

read_args_from_files() (in module lorien.configs) Records (class in lorien.tune.result) RecordToMetadata (class in lorien.dialect.tvm_dial.auto_scheduler_dial.job) register_as_worker() (lorien.tune.rpc.client.RPCClient method) register_config_parser() (in module lorien.configs) register_extractor() (in module lorien.dialect.tvm_dial.frontend_parser) register_job_manager() (in module lorien.tune.job_manager) register_worker() (lorien.tune.rpc.server.RPCService method) relaunch_hanging_jobs() (lorien.tune.job_manager.AWSBatchJobManager method) replay_trace() (lorien.tune.job_manager.JobManagerBase method) request_job() (lorien.tune.rpc.client.RPCClient method) (lorien.tune.rpc.server.RPCService method)

resume_job_states() (lorien.tune.job_manager.AWSBatchJobManager method) (lorien.tune.job_manager.JobManagerBase method) (lorien.tune.job_manager.LocalJobManager method) (lorien.tune.job_manager.RPCJobManager method) RPCClient (class in lorien.tune.rpc.client) RPCJobManager (class in lorien.tune.job_manager) RPCService (class in lorien.tune.rpc.server) run() (in module lorien.tune.master) (lorien.tune.rpc.launch.ClientThread method) run_job_manager() (in module lorien.tune.master)

S

scan_table() (in module lorien.database.table) send_result() (lorien.tune.rpc.client.RPCClient method) (lorien.tune.rpc.server.RPCService method) serialize_framework_build_config() (in module lorien.util)

set_metadata() (lorien.tune.job.Job method) split_s3_path() (in module lorien.util) stateless_hash (lorien.tune.job.Job property) submit() (lorien.tune.rpc.client.RPCClient method)

T

to_job() (lorien.dialect.tvm_dial.auto_scheduler_dial.workload.AutoSchedulerWorkload method) (lorien.dialect.tvm_dial.autotvm_dial.workload.AutoTVMWorkload method) (lorien.workload.Workload method) to_list() (lorien.dialect.tvm_dial.auto_scheduler_dial.result.AutoSchedulerRecords method) (lorien.dialect.tvm_dial.autotvm_dial.result.AutoTVMRecords method) (lorien.dialect.tvm_dial.result.TVMRecords method) (lorien.tune.result.Records method) to_task() (lorien.dialect.tvm_dial.auto_scheduler_dial.workload.AutoSchedulerWorkload method) (lorien.dialect.tvm_dial.autotvm_dial.workload.AutoTVMWorkload method) (lorien.workload.Workload method) trace() (lorien.tune.job.Job method) tune() (lorien.dialect.tvm_dial.auto_scheduler_dial.job.AutoSchedulerJob method) (lorien.dialect.tvm_dial.autotvm_dial.job.AutoTVMJob method) (lorien.dialect.tvm_dial.job.TVMJob method) (lorien.tune.job.Job method) (lorien.tune.job_manager.JobManagerBase method)

tune_impl() (lorien.tune.job_manager.AWSBatchJobManager method) (lorien.tune.job_manager.JobManagerBase method) (lorien.tune.job_manager.LocalJobManager method) (lorien.tune.job_manager.RPCJobManager method) TuneErrorCode (class in lorien.tune.result) TuneMetadata (class in lorien.dialect.tvm_dial.job) TuneResult (class in lorien.tune.result) TVMJob (class in lorien.dialect.tvm_dial.job) TVMJobConfigs (class in lorien.dialect.tvm_dial.job) TVMRecords (class in lorien.dialect.tvm_dial.result)

U

upload_s3_file() (in module lorien.util)

W

Workload (class in lorien.workload)