Index _ | A | B | C | D | E | F | G | H | I | L | M | N | O | P | R | S | T | U | V | W | Z _ __call__() (hidet.graph.FlowGraph method) (hidet.graph.Module method) A abs() (in module hidet.ops) acos() (in module hidet.ops) acosh() (in module hidet.ops) adaptive_avg_pool1d() (in module hidet.ops) adaptive_avg_pool2d() (in module hidet.ops) adaptive_avg_pool3d() (in module hidet.ops) adaptive_max_pool1d() (in module hidet.ops) adaptive_max_pool2d() (in module hidet.ops) adaptive_max_pool3d() (in module hidet.ops) add() (in module hidet.ops) add_edge() (hidet.utils.DirectedGraph method) add_node() (hidet.utils.DirectedGraph method) add_quantize_rules() (hidet.graph.PassContext method) addr() (hidet.option.compile_server static method) after_all_passes() (hidet.graph.GraphPassInstrument method) after_pass() (hidet.graph.GraphPassInstrument method) all() (in module hidet.ops) all_gather() (in module hidet.ops) all_reduce() (in module hidet.ops) any() (in module hidet.ops) arange() (in module hidet.ops) arch() (hidet.option.cpu static method) (hidet.option.cuda static method) arg_reduce() (in module hidet.ir.compute) argmax() (in module hidet.ops) argmin() (in module hidet.ops) as_strided() (in module hidet.ops) asarray() (in module hidet) (in module hidet.graph) asin() (in module hidet.ops) asinh() (in module hidet.ops) assertions (hidet.ir.task.Task attribute) astype() (hidet.Tensor method) atan() (in module hidet.ops) atan2() (in module hidet.ops) atanh() (in module hidet.ops) attention() (in module hidet.ops) attrs (hidet.graph.Operator attribute) (hidet.ir.task.Task attribute) available() (in module hidet.cuda) avg_pool2d() (in module hidet.ops) avg_pool3d() (in module hidet.ops) B barrier() (in module hidet.ops) batch_norm_infer() (in module hidet.ops) before_all_passes() (hidet.graph.GraphPassInstrument method) before_pass() (hidet.graph.GraphPassInstrument method) bench_config() (in module hidet.option) bitwise_and() (in module hidet.ops) bitwise_invert() (in module hidet.ops) bitwise_left_shift() (in module hidet.ops) bitwise_or() (in module hidet.ops) bitwise_right_shift() (in module hidet.ops) bitwise_xor() (in module hidet.ops) broadcast() (in module hidet.ops) build() (hidet.graph.FlowGraph method) (hidet.ir.task.Task method) build_ir_module() (in module hidet.drivers) build_ir_module_batch() (in module hidet.drivers) build_target (hidet.graph.Operator property) build_task() (in module hidet.drivers) C cache_dir() (in module hidet.option) cache_operator() (in module hidet.option) cast() (in module hidet.ops) cdiv() (in module hidet.utils) ceil() (in module hidet.ops) celu() (in module hidet.ops) clamp() (in module hidet.ops) clip() (in module hidet.ops) compile_server (class in hidet.option) CompiledFunction (class in hidet.runtime) CompiledGraph (class in hidet.runtime) CompiledModule (class in hidet.runtime) CompiledTask (class in hidet.runtime) compute() (in module hidet.ir.compute) compute_capability() (in module hidet.cuda) concat() (in module hidet.ops) configs (hidet.graph.PassContext attribute) conj() (in module hidet.ops) context() (in module hidet.option) contiguous() (hidet.Tensor method) conv1d() (in module hidet.ops) conv1d_gemm() (in module hidet.ops) conv1d_transpose() (in module hidet.ops) conv2d() (in module hidet.ops) conv2d_channel_last() (in module hidet.ops) conv2d_gemm() (in module hidet.ops) conv2d_gemm_fp16() (in module hidet.ops) conv2d_gemm_fp16_channel_last() (in module hidet.ops) conv2d_gemm_image_transform() (in module hidet.ops) conv2d_transpose() (in module hidet.ops) conv2d_transpose_gemm() (in module hidet.ops) conv2d_winograd() (in module hidet.ops) conv3d() (in module hidet.ops) conv3d_gemm() (in module hidet.ops) conv3d_transpose() (in module hidet.ops) conv_pad() (in module hidet.ops) copy() (hidet.Tensor method) copy_async() (hidet.Tensor method) correctness_report() (hidet.graph.frontend.torch.DynamoConfig method) cos() (in module hidet.ops) cosh() (in module hidet.ops) cpu (class in hidet.option) cpu() (hidet.Tensor method) cpu_async() (hidet.Tensor method) create_compiled_app() (in module hidet.runtime) cuda (class in hidet.option) cuda() (hidet.Tensor method) cuda_() (hidet.graph.FlowGraph method) cuda_async() (hidet.Tensor method) cuda_batch_matmul() (in module hidet.ops) cuda_graph() (hidet.graph.FlowGraph method) (hidet.runtime.CompiledGraph method) CudaGraph (class in hidet.cuda.graph) cumsum() (in module hidet.ops) current() (hidet.graph.PassContext class method) (hidet.option.OptionContext static method) current_context() (in module hidet.option) current_device() (in module hidet.cuda) current_stream() (in module hidet.cuda) D data_type() (in module hidet.ir.type) DataType (class in hidet.ir.type) debug_cache_tuning() (in module hidet.option) debug_enable_var_id() (in module hidet.option) debug_show_var_id() (in module hidet.option) debug_show_verbose_flow_graph() (in module hidet.option) debug_strict_broadcast_check() (in module hidet.option) DeclareScope (class in hidet.ir.stmt) default_stream() (in module hidet.cuda) detach() (hidet.Tensor method) device (hidet.graph.Operator property) (hidet.Tensor attribute) device_count() (in module hidet.cuda) device_id() (hidet.cuda.Stream method) DirectedGraph (class in hidet.utils) divide() (in module hidet.ops) dtype (hidet.Tensor attribute) dummy_arguments() (hidet.ir.task.Task method) dump_graph_ir() (hidet.graph.frontend.torch.DynamoConfig method) dump_options() (in module hidet.option) DynamoConfig (class in hidet.graph.frontend.torch) E einsum() (in module hidet.ops) elapsed_time() (hidet.cuda.Event method) embedding_bag() (in module hidet.ops) empty() (in module hidet) (in module hidet.graph) empty_like() (in module hidet) (in module hidet.graph) enable() (hidet.option.compile_server static method) enabled() (hidet.option.compile_server static method) equal() (in module hidet.ops) erf() (in module hidet.ops) error_tolerance() (in module hidet.utils) Event (class in hidet.cuda) execution_mode() (in module hidet.option) exp() (in module hidet.ops) expand_dims() (in module hidet.ops) expm1() (in module hidet.ops) ExternalStream (class in hidet.cuda) F factorize() (in module hidet.utils) fix_gpu_frequency_for_tuning() (in module hidet.option) flatten() (in module hidet.ops) flip() (in module hidet.ops) floor() (in module hidet.ops) FlowGraph (class in hidet.graph) forward() (hidet.graph.FlowGraph method) free() (in module hidet.cuda) free_async() (in module hidet.cuda) free_host() (in module hidet.cuda) from_dlpack() (in module hidet) (in module hidet.graph) from_edges() (hidet.utils.DirectedGraph static method) from_numpy() (in module hidet.graph) from_onnx() (in module hidet.graph.frontend) from_torch() (in module hidet) (in module hidet.graph) (in module hidet.graph.frontend) full() (in module hidet) (in module hidet.graph) (in module hidet.ops) full_like() (in module hidet) (in module hidet.graph) Function (class in hidet.ir.func) functions (hidet.runtime.CompiledModule attribute) fused_operator() (in module hidet.ops) G gather() (in module hidet.ops) gc_disabled() (in module hidet.utils) gcd() (in module hidet.utils) gelu() (in module hidet.ops) generate_arguments() (hidet.ir.task.Task method) get_arch() (hidet.option.cpu static method) (hidet.option.cuda static method) get_arch_pair() (hidet.option.cuda static method) get_attr() (hidet.ir.func.Function method) get_bench_config() (in module hidet.option) get_cache_dir() (in module hidet.option) get_cache_operator() (in module hidet.option) get_execution_mode() (in module hidet.option) get_hexcute_matmul() (in module hidet.option) get_num_local_workers() (in module hidet.option) get_num_workers() (hidet.option.compile_server static method) get_option() (hidet.option.OptionContext method) (in module hidet.option) get_parallel_build() (in module hidet.option) get_parallel_k() (in module hidet.option) get_parallel_tune() (in module hidet.option) get_runtime_check() (in module hidet.option) get_save_lower_ir() (in module hidet.option) get_search_space() (in module hidet.option) GraphPassInstrument (class in hidet.graph) greater() (in module hidet.ops) greater_equal() (in module hidet.ops) group_norm() (in module hidet.ops) H handle() (hidet.cuda.Event method) (hidet.cuda.Stream method) hardshrink() (in module hidet.ops) hardsigmoid() (in module hidet.ops) hardswish() (in module hidet.ops) hardtanh() (in module hidet.ops) has_edge() (hidet.utils.DirectedGraph method) has_node() (hidet.utils.DirectedGraph method) hexcute_matmul() (in module hidet.option) hidet module hidet.bfloat16 (built-in variable) hidet.boolean (built-in variable) hidet.drivers module hidet.ffi module hidet.float16 (built-in variable) hidet.float32 (built-in variable) hidet.float64 (built-in variable) hidet.graph module hidet.int16 (built-in variable) hidet.int32 (built-in variable) hidet.int64 (built-in variable) hidet.int8 (built-in variable) hidet.ir.compute module hidet.ir.expr module hidet.ir.func module hidet.ir.stmt module hidet.ir.task module hidet.ops module hidet.option module hidet.runtime module hidet.testing module hidet.tfloat32 (built-in variable) hidet.uint16 (built-in variable) hidet.uint32 (built-in variable) hidet.uint64 (built-in variable) hidet.uint8 (built-in variable) hidet.utils module hip_() (hidet.graph.FlowGraph method) I if_then_else() (in module hidet.ir.expr) IfThenElse (class in hidet.ir.expr) im2col() (in module hidet.ops) imag() (in module hidet.ops) index_select() (in module hidet.ops) initialize() (in module hidet.utils) inputs (hidet.cuda.graph.CudaGraph property) (hidet.graph.FlowGraph attribute) (hidet.graph.Operator attribute) (hidet.ir.task.Task attribute) instance_norm() (in module hidet.ops) instruments (hidet.graph.PassContext attribute) Int (in module hidet.ir.expr) internal (class in hidet.option) inverse_map (hidet.ir.task.Task attribute) is_fix_gpu_frequency_for_tuning() (in module hidet.option) is_option_exist() (in module hidet.option) is_symbolic() (hidet.Tensor method) is_torch_api_use_example_input_shapes() (hidet.option.internal static method) is_use_torch_stream() (in module hidet.option) isfinite() (in module hidet.ops) isinf() (in module hidet.ops) isnan() (in module hidet.ops) item() (hidet.Tensor method) L latency() (hidet.graph.FlowGraph method) layer_norm() (in module hidet.ops) layout (hidet.Tensor attribute) lcm() (in module hidet.utils) leaky_relu() (in module hidet.ops) less() (in module hidet.ops) less_equal() (in module hidet.ops) linspace() (in module hidet.ops) load() (hidet.graph.FlowGraph static method) load_compiled_app() (in module hidet.runtime) load_compiled_graph() (in module hidet) (in module hidet.runtime) load_compiled_module() (in module hidet.runtime) load_compiled_task() (in module hidet.runtime) log() (in module hidet.ops) log10() (in module hidet.ops) log1p() (in module hidet.ops) log2() (in module hidet.ops) logaddexp() (in module hidet.ops) logical_and() (in module hidet.ops) logical_not() (in module hidet.ops) logical_or() (in module hidet.ops) logical_xor() (in module hidet.ops) logsigmoid() (in module hidet.ops) logsoftmax() (in module hidet.ops) lp_norm() (in module hidet.ops) M make_complex() (in module hidet.ops) malloc() (in module hidet.cuda) malloc_async() (in module hidet.cuda) malloc_host() (in module hidet.cuda) matmul() (in module hidet.ops) matmul_cublas() (in module hidet.ops) matmul_nt() (in module hidet.ops) matmul_x86() (in module hidet.ops) max() (in module hidet.ops) max_pool2d() (in module hidet.ops) max_pool3d() (in module hidet.ops) maximum() (in module hidet.ops) mean() (hidet.Tensor method) (in module hidet.ops) memcpy() (in module hidet.cuda) memcpy_async() (in module hidet.cuda) memory_info() (in module hidet.cuda) memset() (in module hidet.cuda) memset_async() (in module hidet.cuda) meshgrid() (in module hidet.ops) min() (in module hidet.ops) minimum() (in module hidet.ops) mod() (in module hidet.ops) module hidet hidet.drivers hidet.ffi hidet.graph hidet.ir.compute hidet.ir.expr hidet.ir.func hidet.ir.stmt hidet.ir.task hidet.ops hidet.option hidet.runtime hidet.testing hidet.utils Module (class in hidet) (class in hidet.graph) module_dir (hidet.runtime.CompiledModule attribute) multiply() (in module hidet.ops) N name (hidet.graph.Operator attribute) (hidet.ir.task.Task attribute) nbytes (hidet.Tensor attribute) negative() (in module hidet.ops) nodes (hidet.graph.FlowGraph property) not_equal() (in module hidet.ops) num_local_workers() (in module hidet.option) num_workers() (hidet.option.compile_server static method) numpy() (hidet.Tensor method) O ones() (in module hidet) (in module hidet.graph) ones_like() (in module hidet) (in module hidet.graph) op (hidet.Tensor attribute) Operator (class in hidet.graph) OperatorPattern (class in hidet.graph.transforms.subgraph_rewrite) optimize() (in module hidet.graph) OptionContext (class in hidet.option) outputs (hidet.cuda.graph.CudaGraph property) (hidet.graph.FlowGraph attribute) (hidet.graph.Operator attribute) (hidet.ir.task.Task attribute) P pad() (in module hidet.ops) parallel_build() (in module hidet.option) parallel_k() (in module hidet.option) parallel_tune() (in module hidet.option) PassContext (class in hidet.graph) password() (hidet.option.compile_server static method) permute_dims() (in module hidet.ops) port() (hidet.option.compile_server static method) positive() (in module hidet.ops) pow() (in module hidet.ops) prelu() (in module hidet.ops) print_input_graph() (hidet.graph.frontend.torch.DynamoConfig method) prod() (in module hidet.ops) profile() (hidet.runtime.CompiledFunction method) (hidet.runtime.CompiledModule method) (hidet.runtime.CompiledTask method) profile_pass_instrument() (hidet.graph.PassContext method) profiler_start() (in module hidet.cuda) profiler_stop() (in module hidet.cuda) properties() (in module hidet.cuda) R randn() (in module hidet) (in module hidet.graph) randn_like() (in module hidet) (in module hidet.graph) real() (in module hidet.ops) rearrange() (hidet.Tensor method) (in module hidet.ops) reciprocal() (in module hidet.ops) record() (hidet.cuda.Event method) reduce() (in module hidet.ir.compute) reduce_cuda_compile_mem() (hidet.graph.PassContext method) reduce_hip_compile_mem() (hidet.graph.PassContext method) reduce_scatter() (in module hidet.ops) ReduceType (class in hidet.ir.compute) register_resolve_rule() (in module hidet.graph.transforms.resolve_variant) register_rewrite_rule() (in module hidet.graph.transforms.subgraph_rewrite) relu() (in module hidet.ops) relu6() (in module hidet.ops) remainder() (in module hidet.ops) repeat_interleave() (in module hidet.ops) repo() (hidet.option.compile_server static method) reset() (hidet.graph.frontend.torch.DynamoConfig method) reshape() (hidet.Tensor method) (in module hidet.ops) resize2d() (in module hidet.ops) resolve() (hidet.graph.transforms.resolve_variant.ResolveRule method) ResolveRule (class in hidet.graph.transforms.resolve_variant) restore_options() (in module hidet.option) roll() (in module hidet.ops) round() (in module hidet.ops) rsqrt() (in module hidet.ops) run() (hidet.cuda.graph.CudaGraph method) run_async() (hidet.cuda.graph.CudaGraph method) (hidet.runtime.CompiledGraph method) (hidet.runtime.CompiledTask method) runtime_check() (in module hidet.option) S save() (hidet.graph.FlowGraph method) (hidet.runtime.CompiledGraph method) save_compiled_app() (in module hidet.runtime) save_compiled_graph() (in module hidet) (in module hidet.runtime) save_graph_instrument() (hidet.graph.PassContext method) save_lower_ir() (in module hidet.option) scalar_input() (in module hidet.ir.compute) scatter_() (in module hidet.ops) scatter_add_() (in module hidet.ops) script() (in module hidet) search_space() (hidet.graph.frontend.torch.DynamoConfig method) (in module hidet.option) set_device() (in module hidet.cuda) set_mma() (hidet.graph.PassContext method) set_option() (hidet.option.OptionContext method) (in module hidet.option) set_precision() (hidet.graph.PassContext method) set_reduce_precision() (hidet.graph.PassContext method) set_strided_slice() (in module hidet.ops) set_verbose() (hidet.graph.PassContext method) set_weights() (hidet.runtime.CompiledGraph method) shape (hidet.Tensor attribute) share_map (hidet.graph.FlowGraph property) (hidet.graph.Operator attribute) (hidet.ir.task.Task attribute) shared_library (hidet.runtime.CompiledModule attribute) SharedLibrary (class in hidet.ffi) sigmoid() (in module hidet.ops) sign() (in module hidet.ops) signature() (hidet.Tensor method) silu() (in module hidet.ops) sin() (in module hidet.ops) sinh() (in module hidet.ops) size (hidet.Tensor attribute) softmax() (in module hidet.ops) softmin() (in module hidet.ops) softplus() (in module hidet.ops) softshrink() (in module hidet.ops) softsign() (in module hidet.ops) source() (hidet.graph.transforms.subgraph_rewrite.SubgraphRewriteRule method) (hidet.runtime.CompiledModule method) split() (in module hidet.ops) sqrt() (in module hidet.ops) square() (in module hidet.ops) squeeze() (hidet.Tensor method) (in module hidet.ops) std() (in module hidet.ops) steal_weights() (hidet.graph.frontend.torch.DynamoConfig method) storage (hidet.Tensor attribute) Stream (class in hidet.cuda) stream() (in module hidet.cuda) strided_slice() (in module hidet.ops) SubgraphRewriteRule (class in hidet.graph.transforms.subgraph_rewrite) subtract() (in module hidet.ops) sum() (hidet.Tensor method) (in module hidet.ops) symbol() (in module hidet) (in module hidet.graph) symbol_like() (in module hidet) (in module hidet.graph) symbols (hidet.ir.task.Task attribute) symmetric_dequantize() (in module hidet.ops) symmetric_quantize() (in module hidet.ops) synchronize() (hidet.cuda.Event method) (hidet.cuda.Stream method) (in module hidet.cuda) T take() (in module hidet.ops) tan() (in module hidet.ops) tanh() (in module hidet.ops) tanhshrink() (in module hidet.ops) target() (hidet.graph.transforms.subgraph_rewrite.SubgraphRewriteRule method) Task (class in hidet.ir.task) task (hidet.graph.Operator attribute) Tensor (class in hidet) tensor_input() (in module hidet.ir.compute) tensor_slice() (in module hidet.ir.expr) TensorPattern (class in hidet.graph.transforms.subgraph_rewrite) tile() (in module hidet.ops) to() (hidet.Tensor method) to_device() (hidet.Tensor method) tolist() (hidet.Tensor method) topological_order() (hidet.utils.DirectedGraph method) torch() (hidet.Tensor method) torch_api_use_example_input_shapes() (hidet.option.internal static method) trace (hidet.Tensor attribute) trace_from() (in module hidet) (in module hidet.graph) transfer() (in module hidet.ops) transpose() (in module hidet.ops) tri() (in module hidet.ops) tril() (in module hidet.ops) triu() (in module hidet.ops) trunc() (in module hidet.ops) U unsqueeze() (hidet.Tensor method) (in module hidet.ops) usage_count (hidet.graph.FlowGraph property) use_attention() (hidet.graph.frontend.torch.DynamoConfig method) use_cuda_graph() (hidet.graph.frontend.torch.DynamoConfig method) use_fp16() (hidet.graph.frontend.torch.DynamoConfig method) use_fp16_reduction() (hidet.graph.frontend.torch.DynamoConfig method) use_tensor_core() (hidet.graph.frontend.torch.DynamoConfig method) use_torch_stream() (in module hidet.option) username() (hidet.option.compile_server static method) V var() (in module hidet.ops) vcuda_() (hidet.graph.FlowGraph method) vhip_() (hidet.graph.FlowGraph method) view() (in module hidet.ops) W wait_event() (hidet.cuda.Stream method) wait_tensor() (in module hidet.ops) where() (in module hidet.ops) Z zeros() (in module hidet) (in module hidet.graph) zeros_like() (in module hidet) (in module hidet.graph)
