For Users
torch.compile
aiu-smi
aiu-trace-analyzer
clone
SimpleMLP
torch.spyre.is_available()
torch.spyre.device_count()
torch.spyre.current_device()
torch.spyre.set_device()
torch.spyre.is_initialized()
torch.spyre.get_amp_supported_dtype()
torch.spyre.manual_seed()
torch.spyre.manual_seed_all()
torch.spyre.get_rng_state()
torch.spyre.set_rng_state()
torch.spyre.initial_seed()
torch.spyre.Stream
torch.spyre.stream()
torch.spyre.current_stream()
torch.spyre.default_stream()
torch.spyre.synchronize()
torch_spyre.profiler.get_diagnostic_report()
torch.spyre.get_diagnostic_report()
torch_spyre.model_utils.load_model_to_spyre()
torch_spyre.model_utils.load_fp8_model_to_spyre()
torch_spyre.model_utils.patch_module_to_for_spyre()
torch_spyre._C.SpyreTensorLayout
torch_spyre._C.DataFormats
torch_spyre._C.get_spyre_tensor_layout()
torch_spyre._C.set_spyre_tensor_layout()
torch_spyre._C.get_downcast_warning()
torch_spyre._C.set_downcast_warning()
torch_spyre.constants.DEVICE_NAME
torch_spyre.constants.DISTRIBUTED_BACKEND_NAME
For Developers
add
indirect_sizes
IndirectAccess
for_each_tile
y = a + b; z = y * c
loop_info
graph.operations
bundle.mlir
ir.Operation
Loops
coarse_tile_pre_stickify
coarse_tile_post_stickify
reorder_unhinted_interlopers
CustomPreSchedulingPasses
CountedLoopSchedulerNode
FusedSchedulerNode
LoopSpec
OpSpec.tiled_symbols
op_spec.py
loop_group_id
codegen_kernel()
scf.for
_divide_ranges
_patch_retiled_load_indexes
MutationLayoutSHOULDREMOVE
propagate_spyre_tensor_layouts
optimize_restickify_locations
finalize_layouts
insert_restickify
insert_post_mutation_restickify
reorder_nonstick_dims
re-fetch
span_reduction
cost_model_matmul_division
work_distribution
rotate
spyreccl
--disable-gil
is_power_z_dev
Explorer