├── .cargo ├── audit.toml └── config.toml ├── .github ├── pull_request_template.md └── workflows │ ├── ci.yml │ └── publish.yml ├── .gitignore ├── .vscode └── settings.json ├── CONTRIBUTING.md ├── Cargo.toml ├── LICENSE-APACHE ├── LICENSE-MIT ├── NOTICE.md ├── README.md ├── assets ├── CubeCL.webp ├── cubecl.drawio.svg └── logo.drawio.svg ├── crates ├── cubecl-common │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ ├── build.rs │ └── src │ │ ├── backtrace.rs │ │ ├── benchmark.rs │ │ ├── bytes │ │ ├── base.rs │ │ ├── default_controller.rs │ │ ├── file.rs │ │ └── mod.rs │ │ ├── cache.rs │ │ ├── cache_file.rs │ │ ├── device.rs │ │ ├── float │ │ ├── fp4.rs │ │ ├── fp6.rs │ │ ├── fp8 │ │ │ ├── fp8_e4m3.rs │ │ │ ├── fp8_e5m2.rs │ │ │ ├── fp8_e8m0.rs │ │ │ └── mod.rs │ │ ├── mod.rs │ │ ├── relaxed.rs │ │ └── tensor_float.rs │ │ ├── format.rs │ │ ├── future.rs │ │ ├── kernel.rs │ │ ├── lib.rs │ │ ├── map.rs │ │ ├── profile.rs │ │ ├── quant │ │ ├── mod.rs │ │ └── scheme.rs │ │ ├── rand.rs │ │ ├── reader.rs │ │ ├── stream_id.rs │ │ └── stub.rs ├── cubecl-core │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── src │ │ ├── codegen │ │ │ ├── compiler.rs │ │ │ ├── integrator.rs │ │ │ ├── metadata.rs │ │ │ └── mod.rs │ │ ├── compute │ │ │ ├── builder.rs │ │ │ ├── launcher.rs │ │ │ └── mod.rs │ │ ├── frontend │ │ │ ├── barrier.rs │ │ │ ├── base.rs │ │ │ ├── branch.rs │ │ │ ├── cmma.rs │ │ │ ├── comptime_error.rs │ │ │ ├── const_expand.rs │ │ │ ├── container │ │ │ │ ├── array │ │ │ │ │ ├── base.rs │ │ │ │ │ ├── launch.rs │ │ │ │ │ └── mod.rs │ │ │ │ ├── base.rs │ │ │ │ ├── cell │ │ │ │ │ ├── comptime.rs │ │ │ │ │ ├── mod.rs │ │ │ │ │ └── runtime.rs │ │ │ │ ├── iter.rs │ │ │ │ ├── line │ │ │ │ │ ├── base.rs │ │ │ │ │ ├── mod.rs │ │ │ │ │ └── ops.rs │ │ │ │ ├── mod.rs │ │ │ │ ├── registry │ │ │ │ │ ├── base.rs │ │ │ │ │ └── mod.rs │ │ │ │ ├── sequence │ │ │ │ │ ├── base.rs │ │ │ │ │ ├── launch.rs │ │ │ │ │ └── mod.rs │ │ │ │ ├── shared_memory.rs │ │ │ │ ├── slice │ │ │ │ │ ├── base.rs │ │ │ │ │ ├── mod.rs │ │ │ │ │ └── operator.rs │ │ │ │ └── tensor │ │ │ │ │ ├── base.rs │ │ │ │ │ ├── launch.rs │ │ │ │ │ ├── mod.rs │ │ │ │ │ └── tensormap.rs │ │ │ ├── debug.rs │ │ │ ├── element │ │ │ │ ├── atomic.rs │ │ │ │ ├── base.rs │ │ │ │ ├── bool.rs │ │ │ │ ├── cast.rs │ │ │ │ ├── cube_elem.rs │ │ │ │ ├── float.rs │ │ │ │ ├── float │ │ │ │ │ ├── fp4.rs │ │ │ │ │ ├── fp6.rs │ │ │ │ │ ├── fp8.rs │ │ │ │ │ ├── relaxed.rs │ │ │ │ │ ├── tensor_float.rs │ │ │ │ │ └── typemap.rs │ │ │ │ ├── int.rs │ │ │ │ ├── int │ │ │ │ │ └── typemap.rs │ │ │ │ ├── mod.rs │ │ │ │ ├── numeric.rs │ │ │ │ └── uint.rs │ │ │ ├── indexation.rs │ │ │ ├── list.rs │ │ │ ├── mod.rs │ │ │ ├── operation │ │ │ │ ├── assignation.rs │ │ │ │ ├── base.rs │ │ │ │ ├── binary.rs │ │ │ │ ├── branch.rs │ │ │ │ ├── clamp.rs │ │ │ │ ├── cmp.rs │ │ │ │ ├── copy.rs │ │ │ │ ├── fma.rs │ │ │ │ ├── mod.rs │ │ │ │ └── unary.rs │ │ │ ├── options.rs │ │ │ ├── plane.rs │ │ │ ├── polyfills.rs │ │ │ ├── synchronization.rs │ │ │ ├── topology.rs │ │ │ └── validation.rs │ │ ├── id.rs │ │ ├── io.rs │ │ ├── lib.rs │ │ ├── pod.rs │ │ ├── post_processing │ │ │ ├── checked_io.rs │ │ │ ├── mod.rs │ │ │ ├── predicate.rs │ │ │ ├── saturating.rs │ │ │ └── unroll.rs │ │ ├── prelude.rs │ │ └── runtime_tests │ │ │ ├── assign.rs │ │ │ ├── atomic.rs │ │ │ ├── barrier.rs │ │ │ ├── binary.rs │ │ │ ├── branch.rs │ │ │ ├── cluster.rs │ │ │ ├── cmma.rs │ │ │ ├── comparison.rs │ │ │ ├── const_match.rs │ │ │ ├── constants.rs │ │ │ ├── debug.rs │ │ │ ├── different_rank.rs │ │ │ ├── enums.rs │ │ │ ├── file.rs │ │ │ ├── index.rs │ │ │ ├── launch.rs │ │ │ ├── line.rs │ │ │ ├── metadata.rs │ │ │ ├── minifloat.rs │ │ │ ├── mod.rs │ │ │ ├── numeric.rs │ │ │ ├── plane.rs │ │ │ ├── saturating.rs │ │ │ ├── sequence.rs │ │ │ ├── slice.rs │ │ │ ├── stream.rs │ │ │ ├── synchronization.rs │ │ │ ├── tensor.rs │ │ │ ├── tensormap.rs │ │ │ ├── to_client.rs │ │ │ ├── topology.rs │ │ │ ├── traits.rs │ │ │ ├── unary.rs │ │ │ └── unroll.rs │ └── tests │ │ ├── error │ │ ├── array_variable.rs │ │ └── array_variable.stderr │ │ └── mod.rs ├── cubecl-cpp │ ├── Cargo.toml │ └── src │ │ ├── cuda │ │ ├── arch.rs │ │ ├── convert.rs │ │ ├── dialect.rs │ │ ├── extension.rs │ │ ├── mma │ │ │ ├── cuda_compiler.rs │ │ │ ├── mod.rs │ │ │ └── ptx_wmma_compiler.rs │ │ ├── mod.rs │ │ ├── processors.rs │ │ └── ptx │ │ │ ├── copy_async.cuh │ │ │ ├── mma.rs │ │ │ ├── mod.rs │ │ │ └── tma_load_im2col.cuh │ │ ├── formatter.rs │ │ ├── hip │ │ ├── arch.rs │ │ ├── dialect.rs │ │ ├── extension.rs │ │ ├── mma │ │ │ ├── mod.rs │ │ │ ├── rocwmma_compiler.rs │ │ │ └── wmma_intrinsics_compiler.rs │ │ ├── mod.rs │ │ └── processors.rs │ │ ├── lib.rs │ │ ├── metal │ │ ├── address_space.rs │ │ ├── arch.rs │ │ ├── attribute.rs │ │ ├── binding.rs │ │ ├── dialect.rs │ │ ├── extension.rs │ │ ├── mod.rs │ │ └── variable.rs │ │ └── shared │ │ ├── barrier.rs │ │ ├── base.rs │ │ ├── binary.rs │ │ ├── body.rs │ │ ├── dialect.rs │ │ ├── element.rs │ │ ├── instruction.rs │ │ ├── item.rs │ │ ├── kernel.rs │ │ ├── mma.rs │ │ ├── mod.rs │ │ ├── pipeline.rs │ │ ├── unary.rs │ │ ├── variable.rs │ │ └── warp.rs ├── cubecl-cpu │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ ├── build.rs │ └── src │ │ ├── compiler │ │ ├── builtin.rs │ │ ├── external_function.rs │ │ ├── memref.rs │ │ ├── mlir_data.rs │ │ ├── mlir_engine.rs │ │ ├── mod.rs │ │ ├── module.rs │ │ ├── passes │ │ │ ├── erf_transform.rs │ │ │ ├── mod.rs │ │ │ └── shared_memories.rs │ │ └── visitor │ │ │ ├── args_manager.rs │ │ │ ├── block.rs │ │ │ ├── elem.rs │ │ │ ├── instruction.rs │ │ │ ├── item.rs │ │ │ ├── mod.rs │ │ │ ├── operation │ │ │ ├── arithmetic.rs │ │ │ ├── bitwise.rs │ │ │ ├── comparison.rs │ │ │ ├── metadata.rs │ │ │ ├── mod.rs │ │ │ ├── operator.rs │ │ │ └── synchronization.rs │ │ │ ├── prelude.rs │ │ │ └── variables.rs │ │ ├── compute │ │ ├── alloc_controller.rs │ │ ├── compute_task.rs │ │ ├── mod.rs │ │ ├── queue.rs │ │ ├── runner.rs │ │ ├── schedule.rs │ │ ├── server.rs │ │ ├── stream.rs │ │ └── worker.rs │ │ ├── device.rs │ │ ├── lib.rs │ │ └── runtime.rs ├── cubecl-cuda │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ ├── build.rs │ └── src │ │ ├── compute │ │ ├── command.rs │ │ ├── context.rs │ │ ├── io │ │ │ ├── controller.rs │ │ │ └── mod.rs │ │ ├── mod.rs │ │ ├── server.rs │ │ ├── storage │ │ │ ├── cpu.rs │ │ │ ├── gpu.rs │ │ │ └── mod.rs │ │ ├── stream.rs │ │ └── sync │ │ │ ├── fence.rs │ │ │ └── mod.rs │ │ ├── device.rs │ │ ├── lib.rs │ │ └── runtime.rs ├── cubecl-hip │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ └── src │ │ ├── compute │ │ ├── command.rs │ │ ├── context.rs │ │ ├── fence.rs │ │ ├── io │ │ │ ├── controller.rs │ │ │ └── mod.rs │ │ ├── mod.rs │ │ ├── server.rs │ │ ├── storage │ │ │ ├── cpu.rs │ │ │ ├── gpu.rs │ │ │ └── mod.rs │ │ └── stream.rs │ │ ├── device.rs │ │ ├── lib.rs │ │ └── runtime.rs ├── cubecl-ir │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ └── src │ │ ├── allocator.rs │ │ ├── arithmetic.rs │ │ ├── atomic.rs │ │ ├── barrier.rs │ │ ├── bitwise.rs │ │ ├── branch.rs │ │ ├── cmma.rs │ │ ├── comparison.rs │ │ ├── lib.rs │ │ ├── marker.rs │ │ ├── metadata.rs │ │ ├── non_semantic.rs │ │ ├── operation.rs │ │ ├── operator.rs │ │ ├── plane.rs │ │ ├── processing.rs │ │ ├── reflect.rs │ │ ├── runtime_properties.rs │ │ ├── scope.rs │ │ ├── synchronization.rs │ │ ├── tma.rs │ │ ├── type.rs │ │ ├── type_hash.rs │ │ └── variable.rs ├── cubecl-macros-internal │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ └── src │ │ ├── generate │ │ ├── mod.rs │ │ ├── op_args.rs │ │ └── operation.rs │ │ ├── lib.rs │ │ ├── parse │ │ ├── mod.rs │ │ ├── op_args.rs │ │ └── operation.rs │ │ └── type_hash.rs ├── cubecl-macros │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── build.rs │ └── src │ │ ├── error.rs │ │ ├── expression.rs │ │ ├── generate │ │ ├── autotune.rs │ │ ├── comptime.rs │ │ ├── cube_impl.rs │ │ ├── cube_trait.rs │ │ ├── cube_type │ │ │ ├── generate.rs │ │ │ ├── generate_enum.rs │ │ │ ├── generate_struct.rs │ │ │ └── mod.rs │ │ ├── expression.rs │ │ ├── kernel.rs │ │ ├── launch.rs │ │ ├── mod.rs │ │ └── statement.rs │ │ ├── lib.rs │ │ ├── operator.rs │ │ ├── parse │ │ ├── autotune.rs │ │ ├── branch.rs │ │ ├── cube_impl.rs │ │ ├── cube_trait.rs │ │ ├── cube_type │ │ │ ├── mod.rs │ │ │ ├── parse.rs │ │ │ ├── parse_enum.rs │ │ │ └── parse_struct.rs │ │ ├── desugar.rs │ │ ├── expression.rs │ │ ├── helpers.rs │ │ ├── kernel.rs │ │ ├── mod.rs │ │ ├── operator.rs │ │ └── statement.rs │ │ ├── paths.rs │ │ ├── scope.rs │ │ └── statement.rs ├── cubecl-opt │ ├── Cargo.toml │ ├── README.md │ └── src │ │ ├── analyses │ │ ├── base.rs │ │ ├── dominance.rs │ │ ├── integer_range.rs │ │ ├── liveness.rs │ │ ├── mod.rs │ │ ├── post_order.rs │ │ ├── uniformity.rs │ │ └── writes.rs │ │ ├── block.rs │ │ ├── control_flow.rs │ │ ├── debug.rs │ │ ├── gvn │ │ ├── analysis.rs │ │ ├── apply.rs │ │ ├── base.rs │ │ ├── convert.rs │ │ ├── mod.rs │ │ └── numbering.rs │ │ ├── instructions.rs │ │ ├── lib.rs │ │ ├── passes │ │ ├── array_copy_propagate.rs │ │ ├── composite.rs │ │ ├── constant_prop.rs │ │ ├── dead_code.rs │ │ ├── expression_merge.rs │ │ ├── index_merge.rs │ │ ├── inlined_if_to_select.rs │ │ ├── mod.rs │ │ └── reduce_strength.rs │ │ ├── phi_frontiers.rs │ │ ├── transformers.rs │ │ └── version.rs ├── cubecl-runtime │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ ├── benches │ │ └── dynamic.rs │ ├── build.rs │ ├── src │ │ ├── client.rs │ │ ├── compiler.rs │ │ ├── config │ │ │ ├── autotune.rs │ │ │ ├── base.rs │ │ │ ├── cache.rs │ │ │ ├── compilation.rs │ │ │ ├── logger.rs │ │ │ ├── memory.rs │ │ │ ├── mod.rs │ │ │ ├── profiling.rs │ │ │ └── streaming.rs │ │ ├── feature_set.rs │ │ ├── features.rs │ │ ├── id.rs │ │ ├── kernel.rs │ │ ├── lib.rs │ │ ├── logging │ │ │ ├── mod.rs │ │ │ ├── profiling.rs │ │ │ └── server.rs │ │ ├── memory_management │ │ │ ├── base.rs │ │ │ ├── memory_manage.rs │ │ │ ├── memory_pool │ │ │ │ ├── base.rs │ │ │ │ ├── exclusive_pool.rs │ │ │ │ ├── handle.rs │ │ │ │ ├── memory_page.rs │ │ │ │ ├── mod.rs │ │ │ │ ├── persistent_pool.rs │ │ │ │ └── sliced_pool.rs │ │ │ └── mod.rs │ │ ├── runtime.rs │ │ ├── server.rs │ │ ├── storage │ │ │ ├── base.rs │ │ │ ├── bytes_cpu.rs │ │ │ └── mod.rs │ │ ├── stream │ │ │ ├── base.rs │ │ │ ├── event.rs │ │ │ ├── mod.rs │ │ │ └── scheduler.rs │ │ ├── timestamp_profiler.rs │ │ ├── tma.rs │ │ └── tune │ │ │ ├── base.rs │ │ │ ├── function_tunable.rs │ │ │ ├── input_generator.rs │ │ │ ├── key_generator.rs │ │ │ ├── local.rs │ │ │ ├── mod.rs │ │ │ ├── operation.rs │ │ │ ├── tune_benchmark.rs │ │ │ ├── tune_cache.rs │ │ │ ├── tuner.rs │ │ │ └── util.rs │ └── tests │ │ ├── dummy │ │ ├── compute.rs │ │ ├── kernel.rs │ │ ├── mod.rs │ │ ├── server.rs │ │ └── tune │ │ │ ├── autotune_operations.rs │ │ │ ├── kernels.rs │ │ │ ├── mod.rs │ │ │ └── operation_sets.rs │ │ └── integration_test.rs ├── cubecl-spirv │ ├── Cargo.toml │ ├── README.md │ └── src │ │ ├── arithmetic.rs │ │ ├── atomic.rs │ │ ├── bitwise.rs │ │ ├── branch.rs │ │ ├── cmma.rs │ │ ├── compiler.rs │ │ ├── debug.rs │ │ ├── extensions.rs │ │ ├── globals.rs │ │ ├── instruction.rs │ │ ├── item.rs │ │ ├── lib.rs │ │ ├── lookups.rs │ │ ├── metadata.rs │ │ ├── subgroup.rs │ │ ├── sync.rs │ │ ├── target.rs │ │ ├── transformers.rs │ │ └── variable.rs ├── cubecl-std │ ├── Cargo.toml │ └── src │ │ ├── event │ │ └── mod.rs │ │ ├── fast_math.rs │ │ ├── lib.rs │ │ ├── option.rs │ │ ├── quant │ │ ├── base.rs │ │ ├── dequantize.rs │ │ ├── mod.rs │ │ └── view.rs │ │ ├── reinterpret_slice.rs │ │ ├── scalar.rs │ │ ├── swizzle.rs │ │ ├── tensor │ │ ├── contiguous.rs │ │ ├── handle.rs │ │ ├── identity.rs │ │ ├── layout │ │ │ ├── as_dyn.rs │ │ │ ├── base.rs │ │ │ ├── chain.rs │ │ │ ├── coordinates.rs │ │ │ ├── linear.rs │ │ │ ├── mod.rs │ │ │ ├── permuted.rs │ │ │ ├── plain.rs │ │ │ ├── simple.rs │ │ │ ├── slice.rs │ │ │ ├── strided.rs │ │ │ └── virtual.rs │ │ ├── matrix_batch_layout.rs │ │ ├── mod.rs │ │ ├── view │ │ │ ├── as_view.rs │ │ │ ├── base.rs │ │ │ ├── idx.rs │ │ │ ├── launch.rs │ │ │ ├── mod.rs │ │ │ └── operations │ │ │ │ ├── array.rs │ │ │ │ ├── base.rs │ │ │ │ ├── mod.rs │ │ │ │ ├── slice.rs │ │ │ │ ├── tensor_map.rs │ │ │ │ ├── view.rs │ │ │ │ ├── virtual_tensor.rs │ │ │ │ └── virtual_view.rs │ │ └── virtual.rs │ │ ├── tests │ │ ├── event.rs │ │ ├── mod.rs │ │ ├── reinterpret_slice.rs │ │ ├── tensor │ │ │ ├── identity.rs │ │ │ ├── mod.rs │ │ │ ├── test_macros │ │ │ │ ├── identity.rs │ │ │ │ └── mod.rs │ │ │ └── test_utils.rs │ │ ├── trigonometry.rs │ │ └── view │ │ │ ├── mod.rs │ │ │ └── quantized.rs │ │ └── trigonometry.rs ├── cubecl-wgpu │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── README.md │ ├── build.rs │ └── src │ │ ├── backend │ │ ├── base.rs │ │ ├── metal.rs │ │ ├── mod.rs │ │ ├── vulkan.rs │ │ ├── vulkan │ │ │ └── features.rs │ │ └── wgsl.rs │ │ ├── compiler │ │ ├── base.rs │ │ ├── mod.rs │ │ └── wgsl │ │ │ ├── base.rs │ │ │ ├── body.rs │ │ │ ├── compiler.rs │ │ │ ├── extension.rs │ │ │ ├── instructions.rs │ │ │ ├── mod.rs │ │ │ ├── shader.rs │ │ │ └── subgroup.rs │ │ ├── compute │ │ ├── controller.rs │ │ ├── errors.rs │ │ ├── mem_manager.rs │ │ ├── mod.rs │ │ ├── poll.rs │ │ ├── schedule.rs │ │ ├── server.rs │ │ ├── storage.rs │ │ ├── stream.rs │ │ └── timings.rs │ │ ├── device.rs │ │ ├── element.rs │ │ ├── graphics.rs │ │ ├── lib.rs │ │ └── runtime.rs └── cubecl │ ├── Cargo.toml │ ├── LICENSE-APACHE │ ├── LICENSE-MIT │ ├── build.rs │ └── src │ └── lib.rs ├── cubecl-book ├── .gitignore ├── .prettierrc.json ├── book.toml └── src │ ├── SUMMARY.md │ ├── advanced-usage │ ├── config.md │ ├── math_optimizations.md │ └── summary.md │ ├── core-features │ ├── autotune.md │ ├── comptime.md │ ├── features.md │ ├── summary.md │ └── vectorization.md │ ├── getting-started │ ├── benchmark.md │ ├── cubecl.drawio.svg │ ├── examples.md │ ├── installation.md │ ├── parallel_reduction.md │ ├── parallel_reduction_3d.md │ ├── simple_reduction.md │ ├── src │ │ ├── bin │ │ │ ├── v1-cpu.rs │ │ │ ├── v2-gpu.rs │ │ │ ├── v3-gpu.rs │ │ │ ├── v4-gpu.rs │ │ │ ├── v5-gpu.rs │ │ │ ├── v6-gpu.rs │ │ │ └── v7-gpu.rs │ │ ├── cpu_tensor.rs │ │ ├── gpu_tensor.rs │ │ └── lib.rs │ ├── summary.md │ └── vectorized_reduction.md │ ├── language-support │ ├── enum.md │ ├── struct.md │ ├── summary.md │ └── trait.md │ ├── motivation.md │ └── overview.md ├── examples ├── device_sharing │ ├── Cargo.toml │ ├── examples │ │ └── device_sharing.rs │ └── src │ │ └── lib.rs ├── fusing │ ├── Cargo.toml │ ├── examples │ │ └── fusing.rs │ └── src │ │ └── lib.rs ├── gelu │ ├── Cargo.toml │ ├── examples │ │ └── gelu.rs │ └── src │ │ └── lib.rs ├── normalization │ ├── Cargo.toml │ ├── examples │ │ └── normalization.rs │ └── src │ │ └── lib.rs └── sum_things │ ├── Cargo.toml │ ├── examples │ └── sum_things.rs │ └── src │ └── lib.rs ├── typos.toml └── xtask ├── Cargo.toml ├── README.md └── src ├── commands ├── book.rs ├── build.rs ├── check.rs ├── mod.rs ├── profile.rs ├── test.rs └── validate.rs ├── main.rs └── utils ├── cargo.rs ├── mod.rs ├── process.rs ├── prompt.rs ├── time.rs └── workspace.rs /.cargo/audit.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/.cargo/audit.toml -------------------------------------------------------------------------------- /.cargo/config.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/.cargo/config.toml -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- 1 | { 2 | "files.eol": "\n" 3 | } 4 | -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/Cargo.toml -------------------------------------------------------------------------------- /LICENSE-APACHE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/LICENSE-APACHE -------------------------------------------------------------------------------- /LICENSE-MIT: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/LICENSE-MIT -------------------------------------------------------------------------------- /NOTICE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/NOTICE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/README.md -------------------------------------------------------------------------------- /assets/CubeCL.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/assets/CubeCL.webp -------------------------------------------------------------------------------- /assets/cubecl.drawio.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/assets/cubecl.drawio.svg -------------------------------------------------------------------------------- /assets/logo.drawio.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/assets/logo.drawio.svg -------------------------------------------------------------------------------- /crates/cubecl-common/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-common/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-common/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-common/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/README.md -------------------------------------------------------------------------------- /crates/cubecl-common/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/build.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/backtrace.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/backtrace.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/benchmark.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/benchmark.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/bytes/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/bytes/base.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/bytes/default_controller.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/bytes/default_controller.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/bytes/file.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/bytes/file.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/bytes/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/bytes/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/cache.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/cache.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/cache_file.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/cache_file.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/device.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/device.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/fp4.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/fp4.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/fp6.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/fp6.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/fp8/fp8_e4m3.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/fp8/fp8_e4m3.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/fp8/fp8_e5m2.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/fp8/fp8_e5m2.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/fp8/fp8_e8m0.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/fp8/fp8_e8m0.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/fp8/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/fp8/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/relaxed.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/relaxed.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/float/tensor_float.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/float/tensor_float.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/format.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/format.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/future.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/future.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/kernel.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/kernel.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/map.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/map.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/profile.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/profile.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/quant/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/quant/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/quant/scheme.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/quant/scheme.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/rand.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/rand.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/reader.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/reader.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/stream_id.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/stream_id.rs -------------------------------------------------------------------------------- /crates/cubecl-common/src/stub.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-common/src/stub.rs -------------------------------------------------------------------------------- /crates/cubecl-core/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-core/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-core/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-core/src/codegen/compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/codegen/compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/codegen/integrator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/codegen/integrator.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/codegen/metadata.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/codegen/metadata.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/codegen/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/codegen/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/compute/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/compute/builder.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/compute/launcher.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/compute/launcher.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/compute/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/compute/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/barrier.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/barrier.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/branch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/branch.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/cmma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/cmma.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/comptime_error.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/comptime_error.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/const_expand.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/const_expand.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/array/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/array/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/array/launch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/array/launch.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/array/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/array/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/cell/comptime.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/cell/comptime.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/cell/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/cell/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/cell/runtime.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/cell/runtime.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/iter.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/iter.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/line/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/line/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/line/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/line/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/line/ops.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/line/ops.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/registry/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/registry/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/registry/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/registry/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/sequence/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/sequence/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/sequence/launch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/sequence/launch.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/sequence/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/sequence/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/shared_memory.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/shared_memory.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/slice/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/slice/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/slice/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/slice/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/slice/operator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/slice/operator.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/tensor/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/tensor/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/tensor/launch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/tensor/launch.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/tensor/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/tensor/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/container/tensor/tensormap.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/container/tensor/tensormap.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/debug.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/debug.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/atomic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/atomic.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/bool.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/bool.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/cast.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/cast.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/cube_elem.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/cube_elem.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/float.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/float.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/float/fp4.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/float/fp4.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/float/fp6.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/float/fp6.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/float/fp8.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/float/fp8.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/float/relaxed.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/float/relaxed.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/float/tensor_float.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/float/tensor_float.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/float/typemap.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/float/typemap.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/int.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/int.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/int/typemap.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/int/typemap.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/numeric.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/numeric.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/element/uint.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/element/uint.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/indexation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/indexation.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/list.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/list.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/assignation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/assignation.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/base.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/binary.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/binary.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/branch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/branch.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/clamp.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/clamp.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/cmp.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/cmp.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/copy.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/copy.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/fma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/fma.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/operation/unary.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/operation/unary.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/options.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/options.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/plane.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/plane.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/polyfills.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/polyfills.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/synchronization.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/synchronization.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/topology.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/topology.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/frontend/validation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/frontend/validation.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/id.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/id.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/io.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/io.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/pod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/pod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/post_processing/checked_io.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/post_processing/checked_io.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/post_processing/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/post_processing/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/post_processing/predicate.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/post_processing/predicate.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/post_processing/saturating.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/post_processing/saturating.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/post_processing/unroll.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/post_processing/unroll.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/prelude.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/prelude.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/assign.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/assign.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/atomic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/atomic.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/barrier.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/barrier.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/binary.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/binary.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/branch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/branch.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/cluster.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/cluster.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/cmma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/cmma.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/comparison.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/comparison.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/const_match.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/const_match.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/constants.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/constants.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/debug.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/debug.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/different_rank.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/different_rank.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/enums.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/enums.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/file.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/file.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/index.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/index.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/launch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/launch.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/line.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/line.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/metadata.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/metadata.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/minifloat.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/minifloat.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/numeric.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/numeric.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/plane.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/plane.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/saturating.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/saturating.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/sequence.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/sequence.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/slice.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/slice.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/stream.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/stream.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/synchronization.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/synchronization.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/tensor.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/tensor.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/tensormap.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/tensormap.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/to_client.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/to_client.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/topology.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/topology.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/traits.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/traits.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/unary.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/unary.rs -------------------------------------------------------------------------------- /crates/cubecl-core/src/runtime_tests/unroll.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/src/runtime_tests/unroll.rs -------------------------------------------------------------------------------- /crates/cubecl-core/tests/error/array_variable.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/tests/error/array_variable.rs -------------------------------------------------------------------------------- /crates/cubecl-core/tests/error/array_variable.stderr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/tests/error/array_variable.stderr -------------------------------------------------------------------------------- /crates/cubecl-core/tests/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-core/tests/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/arch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/arch.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/convert.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/convert.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/dialect.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/dialect.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/extension.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/extension.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/mma/cuda_compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/mma/cuda_compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/mma/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/mma/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/mma/ptx_wmma_compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/mma/ptx_wmma_compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/processors.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/processors.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/ptx/copy_async.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/ptx/copy_async.cuh -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/ptx/mma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/ptx/mma.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/ptx/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/ptx/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/cuda/ptx/tma_load_im2col.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/cuda/ptx/tma_load_im2col.cuh -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/formatter.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/formatter.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/arch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/arch.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/dialect.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/dialect.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/extension.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/extension.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/mma/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/mma/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/mma/rocwmma_compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/mma/rocwmma_compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/mma/wmma_intrinsics_compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/mma/wmma_intrinsics_compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/hip/processors.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/hip/processors.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/address_space.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/address_space.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/arch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/arch.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/attribute.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/attribute.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/binding.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/binding.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/dialect.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/dialect.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/extension.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/extension.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/metal/variable.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/metal/variable.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/barrier.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/barrier.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/base.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/binary.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/binary.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/body.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/body.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/dialect.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/dialect.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/element.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/element.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/instruction.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/instruction.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/item.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/item.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/kernel.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/kernel.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/mma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/mma.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/pipeline.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/unary.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/unary.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/variable.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/variable.rs -------------------------------------------------------------------------------- /crates/cubecl-cpp/src/shared/warp.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpp/src/shared/warp.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-cpu/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-cpu/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-cpu/README.md: -------------------------------------------------------------------------------- 1 | # Cpu runtime 2 | 3 | ## Setup 4 | -------------------------------------------------------------------------------- /crates/cubecl-cpu/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/build.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/builtin.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/builtin.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/external_function.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/external_function.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/memref.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/memref.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/mlir_data.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/mlir_data.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/mlir_engine.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/mlir_engine.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/module.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/module.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/passes/erf_transform.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/passes/erf_transform.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/passes/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/passes/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/passes/shared_memories.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/passes/shared_memories.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/args_manager.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/args_manager.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/block.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/block.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/elem.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/elem.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/instruction.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/instruction.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/item.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/item.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/operation/arithmetic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/operation/arithmetic.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/operation/bitwise.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/operation/bitwise.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/operation/comparison.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/operation/comparison.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/operation/metadata.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/operation/metadata.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/operation/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/operation/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/operation/operator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/operation/operator.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/operation/synchronization.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/operation/synchronization.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/prelude.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/prelude.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compiler/visitor/variables.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compiler/visitor/variables.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/alloc_controller.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/alloc_controller.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/compute_task.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/compute_task.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/queue.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/queue.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/runner.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/runner.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/schedule.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/schedule.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/server.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/server.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/stream.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/stream.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/compute/worker.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/compute/worker.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/device.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/device.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-cpu/src/runtime.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cpu/src/runtime.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-cuda/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-cuda/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-cuda/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/README.md -------------------------------------------------------------------------------- /crates/cubecl-cuda/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/build.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/command.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/command.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/context.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/context.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/io/controller.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/io/controller.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/io/mod.rs: -------------------------------------------------------------------------------- 1 | pub(crate) mod controller; 2 | -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/server.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/server.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/storage/cpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/storage/cpu.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/storage/gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/storage/gpu.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/storage/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/storage/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/stream.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/stream.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/sync/fence.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/sync/fence.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/compute/sync/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/compute/sync/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/device.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/device.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-cuda/src/runtime.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-cuda/src/runtime.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-hip/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-hip/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-hip/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/README.md -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/command.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/command.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/context.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/context.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/fence.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/fence.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/io/controller.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/io/controller.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/io/mod.rs: -------------------------------------------------------------------------------- 1 | pub(crate) mod controller; 2 | -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/server.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/server.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/storage/cpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/storage/cpu.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/storage/gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/storage/gpu.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/storage/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/storage/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/compute/stream.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/compute/stream.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/device.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/device.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-hip/src/runtime.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-hip/src/runtime.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-ir/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-ir/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-ir/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/README.md -------------------------------------------------------------------------------- /crates/cubecl-ir/src/allocator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/allocator.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/arithmetic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/arithmetic.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/atomic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/atomic.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/barrier.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/barrier.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/bitwise.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/bitwise.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/branch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/branch.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/cmma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/cmma.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/comparison.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/comparison.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/marker.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/marker.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/metadata.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/metadata.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/non_semantic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/non_semantic.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/operation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/operation.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/operator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/operator.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/plane.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/plane.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/processing.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/processing.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/reflect.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/reflect.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/runtime_properties.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/runtime_properties.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/scope.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/scope.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/synchronization.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/synchronization.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/tma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/tma.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/type.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/type.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/type_hash.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/type_hash.rs -------------------------------------------------------------------------------- /crates/cubecl-ir/src/variable.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-ir/src/variable.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/README.md -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/generate/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/generate/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/generate/op_args.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/generate/op_args.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/generate/operation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/generate/operation.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/parse/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/parse/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/parse/op_args.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/parse/op_args.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/parse/operation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/parse/operation.rs -------------------------------------------------------------------------------- /crates/cubecl-macros-internal/src/type_hash.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros-internal/src/type_hash.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-macros/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-macros/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-macros/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/build.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/error.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/error.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/expression.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/expression.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/autotune.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/autotune.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/comptime.rs: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/cube_impl.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/cube_impl.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/cube_trait.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/cube_trait.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/cube_type/generate.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/cube_type/generate.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/cube_type/generate_enum.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/cube_type/generate_enum.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/cube_type/generate_struct.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/cube_type/generate_struct.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/cube_type/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/cube_type/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/expression.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/expression.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/kernel.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/kernel.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/launch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/launch.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/generate/statement.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/generate/statement.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/operator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/operator.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/autotune.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/autotune.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/branch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/branch.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/cube_impl.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/cube_impl.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/cube_trait.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/cube_trait.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/cube_type/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/cube_type/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/cube_type/parse.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/cube_type/parse.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/cube_type/parse_enum.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/cube_type/parse_enum.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/cube_type/parse_struct.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/cube_type/parse_struct.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/desugar.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/desugar.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/expression.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/expression.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/helpers.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/helpers.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/kernel.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/kernel.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/operator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/operator.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/parse/statement.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/parse/statement.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/paths.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/paths.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/scope.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/scope.rs -------------------------------------------------------------------------------- /crates/cubecl-macros/src/statement.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-macros/src/statement.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-opt/README.md: -------------------------------------------------------------------------------- 1 | # Compiler Optimizations 2 | 3 | WIP 4 | -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/base.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/dominance.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/dominance.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/integer_range.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/integer_range.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/liveness.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/liveness.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/post_order.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/post_order.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/uniformity.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/uniformity.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/analyses/writes.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/analyses/writes.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/block.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/block.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/control_flow.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/control_flow.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/debug.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/debug.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/gvn/analysis.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/gvn/analysis.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/gvn/apply.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/gvn/apply.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/gvn/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/gvn/base.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/gvn/convert.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/gvn/convert.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/gvn/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/gvn/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/gvn/numbering.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/gvn/numbering.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/instructions.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/instructions.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/array_copy_propagate.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/array_copy_propagate.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/composite.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/composite.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/constant_prop.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/constant_prop.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/dead_code.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/dead_code.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/expression_merge.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/expression_merge.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/index_merge.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/index_merge.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/inlined_if_to_select.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/inlined_if_to_select.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/passes/reduce_strength.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/passes/reduce_strength.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/phi_frontiers.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/phi_frontiers.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/transformers.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/transformers.rs -------------------------------------------------------------------------------- /crates/cubecl-opt/src/version.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-opt/src/version.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-runtime/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-runtime/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-runtime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/README.md -------------------------------------------------------------------------------- /crates/cubecl-runtime/benches/dynamic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/benches/dynamic.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/build.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/client.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/client.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/autotune.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/autotune.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/base.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/cache.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/cache.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/compilation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/compilation.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/logger.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/logger.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/memory.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/memory.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/profiling.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/profiling.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/config/streaming.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/config/streaming.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/feature_set.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/feature_set.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/features.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/features.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/id.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/id.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/kernel.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/kernel.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/logging/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/logging/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/logging/profiling.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/logging/profiling.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/logging/server.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/logging/server.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/base.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_manage.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_manage.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_pool/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_pool/base.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_pool/exclusive_pool.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_pool/exclusive_pool.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_pool/handle.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_pool/handle.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_pool/memory_page.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_pool/memory_page.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_pool/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_pool/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_pool/persistent_pool.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_pool/persistent_pool.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/memory_pool/sliced_pool.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/memory_pool/sliced_pool.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/memory_management/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/memory_management/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/runtime.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/runtime.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/server.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/server.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/storage/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/storage/base.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/storage/bytes_cpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/storage/bytes_cpu.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/storage/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/storage/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/stream/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/stream/base.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/stream/event.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/stream/event.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/stream/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/stream/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/stream/scheduler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/stream/scheduler.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/timestamp_profiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/timestamp_profiler.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tma.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/base.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/function_tunable.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/function_tunable.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/input_generator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/input_generator.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/key_generator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/key_generator.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/local.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/local.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/operation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/operation.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/tune_benchmark.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/tune_benchmark.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/tune_cache.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/tune_cache.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/tuner.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/tuner.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/src/tune/util.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/src/tune/util.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/compute.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/compute.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/kernel.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/kernel.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/server.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/server.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/tune/autotune_operations.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/tune/autotune_operations.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/tune/kernels.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/tune/kernels.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/tune/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/tune/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/dummy/tune/operation_sets.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/dummy/tune/operation_sets.rs -------------------------------------------------------------------------------- /crates/cubecl-runtime/tests/integration_test.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-runtime/tests/integration_test.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-spirv/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/README.md -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/arithmetic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/arithmetic.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/atomic.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/atomic.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/bitwise.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/bitwise.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/branch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/branch.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/cmma.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/cmma.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/debug.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/debug.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/extensions.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/extensions.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/globals.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/globals.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/instruction.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/instruction.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/item.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/item.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/lookups.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/lookups.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/metadata.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/metadata.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/subgroup.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/subgroup.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/sync.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/sync.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/target.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/target.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/transformers.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/transformers.rs -------------------------------------------------------------------------------- /crates/cubecl-spirv/src/variable.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-spirv/src/variable.rs -------------------------------------------------------------------------------- /crates/cubecl-std/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-std/src/event/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/event/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/fast_math.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/fast_math.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/option.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/option.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/quant/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/quant/base.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/quant/dequantize.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/quant/dequantize.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/quant/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/quant/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/quant/view.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/quant/view.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/reinterpret_slice.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/reinterpret_slice.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/scalar.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/scalar.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/swizzle.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/swizzle.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/contiguous.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/contiguous.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/handle.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/handle.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/identity.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/identity.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/as_dyn.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/as_dyn.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/base.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/chain.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/chain.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/coordinates.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/coordinates.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/linear.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/linear.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/permuted.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/permuted.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/plain.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/plain.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/simple.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/simple.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/slice.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/slice.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/strided.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/strided.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/layout/virtual.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/layout/virtual.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/matrix_batch_layout.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/matrix_batch_layout.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/as_view.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/as_view.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/base.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/idx.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/idx.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/launch.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/launch.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/array.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/array.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/base.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/slice.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/slice.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/tensor_map.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/tensor_map.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/view.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/view.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/virtual_tensor.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/virtual_tensor.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/view/operations/virtual_view.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/view/operations/virtual_view.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tensor/virtual.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tensor/virtual.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/event.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/event.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/reinterpret_slice.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/reinterpret_slice.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/tensor/identity.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/tensor/identity.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/tensor/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/tensor/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/tensor/test_macros/identity.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/tensor/test_macros/identity.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/tensor/test_macros/mod.rs: -------------------------------------------------------------------------------- 1 | mod identity; 2 | -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/tensor/test_utils.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/tensor/test_utils.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/trigonometry.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/trigonometry.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/view/mod.rs: -------------------------------------------------------------------------------- 1 | pub mod quantized; 2 | -------------------------------------------------------------------------------- /crates/cubecl-std/src/tests/view/quantized.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/tests/view/quantized.rs -------------------------------------------------------------------------------- /crates/cubecl-std/src/trigonometry.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-std/src/trigonometry.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl-wgpu/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl-wgpu/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl-wgpu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/README.md -------------------------------------------------------------------------------- /crates/cubecl-wgpu/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/build.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/backend/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/backend/base.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/backend/metal.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/backend/metal.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/backend/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/backend/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/backend/vulkan.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/backend/vulkan.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/backend/vulkan/features.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/backend/vulkan/features.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/backend/wgsl.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/backend/wgsl.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/base.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/base.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/body.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/body.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/compiler.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/compiler.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/extension.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/extension.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/instructions.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/instructions.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/shader.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/shader.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compiler/wgsl/subgroup.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compiler/wgsl/subgroup.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/controller.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/controller.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/errors.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/errors.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/mem_manager.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/mem_manager.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/mod.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/poll.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/poll.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/schedule.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/schedule.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/server.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/server.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/storage.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/storage.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/stream.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/stream.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/compute/timings.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/compute/timings.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/device.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/device.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/element.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/element.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/graphics.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/graphics.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/lib.rs -------------------------------------------------------------------------------- /crates/cubecl-wgpu/src/runtime.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl-wgpu/src/runtime.rs -------------------------------------------------------------------------------- /crates/cubecl/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl/Cargo.toml -------------------------------------------------------------------------------- /crates/cubecl/LICENSE-APACHE: -------------------------------------------------------------------------------- 1 | ../../LICENSE-APACHE -------------------------------------------------------------------------------- /crates/cubecl/LICENSE-MIT: -------------------------------------------------------------------------------- 1 | ../../LICENSE-MIT -------------------------------------------------------------------------------- /crates/cubecl/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl/build.rs -------------------------------------------------------------------------------- /crates/cubecl/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/crates/cubecl/src/lib.rs -------------------------------------------------------------------------------- /cubecl-book/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/.gitignore -------------------------------------------------------------------------------- /cubecl-book/.prettierrc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/.prettierrc.json -------------------------------------------------------------------------------- /cubecl-book/book.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/book.toml -------------------------------------------------------------------------------- /cubecl-book/src/SUMMARY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/SUMMARY.md -------------------------------------------------------------------------------- /cubecl-book/src/advanced-usage/config.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/advanced-usage/config.md -------------------------------------------------------------------------------- /cubecl-book/src/advanced-usage/math_optimizations.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/advanced-usage/math_optimizations.md -------------------------------------------------------------------------------- /cubecl-book/src/advanced-usage/summary.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/advanced-usage/summary.md -------------------------------------------------------------------------------- /cubecl-book/src/core-features/autotune.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/core-features/autotune.md -------------------------------------------------------------------------------- /cubecl-book/src/core-features/comptime.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/core-features/comptime.md -------------------------------------------------------------------------------- /cubecl-book/src/core-features/features.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/core-features/features.md -------------------------------------------------------------------------------- /cubecl-book/src/core-features/summary.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/core-features/summary.md -------------------------------------------------------------------------------- /cubecl-book/src/core-features/vectorization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/core-features/vectorization.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/benchmark.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/benchmark.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/cubecl.drawio.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/cubecl.drawio.svg -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/examples.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/examples.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/installation.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/parallel_reduction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/parallel_reduction.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/parallel_reduction_3d.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/parallel_reduction_3d.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/simple_reduction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/simple_reduction.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/bin/v1-cpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/bin/v1-cpu.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/bin/v2-gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/bin/v2-gpu.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/bin/v3-gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/bin/v3-gpu.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/bin/v4-gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/bin/v4-gpu.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/bin/v5-gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/bin/v5-gpu.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/bin/v6-gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/bin/v6-gpu.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/bin/v7-gpu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/bin/v7-gpu.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/cpu_tensor.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/cpu_tensor.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/gpu_tensor.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/gpu_tensor.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/src/lib.rs -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/summary.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/summary.md -------------------------------------------------------------------------------- /cubecl-book/src/getting-started/vectorized_reduction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/getting-started/vectorized_reduction.md -------------------------------------------------------------------------------- /cubecl-book/src/language-support/enum.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/language-support/enum.md -------------------------------------------------------------------------------- /cubecl-book/src/language-support/struct.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/language-support/struct.md -------------------------------------------------------------------------------- /cubecl-book/src/language-support/summary.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/language-support/summary.md -------------------------------------------------------------------------------- /cubecl-book/src/language-support/trait.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/language-support/trait.md -------------------------------------------------------------------------------- /cubecl-book/src/motivation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/motivation.md -------------------------------------------------------------------------------- /cubecl-book/src/overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/cubecl-book/src/overview.md -------------------------------------------------------------------------------- /examples/device_sharing/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/device_sharing/Cargo.toml -------------------------------------------------------------------------------- /examples/device_sharing/examples/device_sharing.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/device_sharing/examples/device_sharing.rs -------------------------------------------------------------------------------- /examples/device_sharing/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/device_sharing/src/lib.rs -------------------------------------------------------------------------------- /examples/fusing/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/fusing/Cargo.toml -------------------------------------------------------------------------------- /examples/fusing/examples/fusing.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/fusing/examples/fusing.rs -------------------------------------------------------------------------------- /examples/fusing/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/fusing/src/lib.rs -------------------------------------------------------------------------------- /examples/gelu/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/gelu/Cargo.toml -------------------------------------------------------------------------------- /examples/gelu/examples/gelu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/gelu/examples/gelu.rs -------------------------------------------------------------------------------- /examples/gelu/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/gelu/src/lib.rs -------------------------------------------------------------------------------- /examples/normalization/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/normalization/Cargo.toml -------------------------------------------------------------------------------- /examples/normalization/examples/normalization.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/normalization/examples/normalization.rs -------------------------------------------------------------------------------- /examples/normalization/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/normalization/src/lib.rs -------------------------------------------------------------------------------- /examples/sum_things/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/sum_things/Cargo.toml -------------------------------------------------------------------------------- /examples/sum_things/examples/sum_things.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/sum_things/examples/sum_things.rs -------------------------------------------------------------------------------- /examples/sum_things/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/examples/sum_things/src/lib.rs -------------------------------------------------------------------------------- /typos.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/typos.toml -------------------------------------------------------------------------------- /xtask/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/Cargo.toml -------------------------------------------------------------------------------- /xtask/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/README.md -------------------------------------------------------------------------------- /xtask/src/commands/book.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/commands/book.rs -------------------------------------------------------------------------------- /xtask/src/commands/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/commands/build.rs -------------------------------------------------------------------------------- /xtask/src/commands/check.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/commands/check.rs -------------------------------------------------------------------------------- /xtask/src/commands/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/commands/mod.rs -------------------------------------------------------------------------------- /xtask/src/commands/profile.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/commands/profile.rs -------------------------------------------------------------------------------- /xtask/src/commands/test.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/commands/test.rs -------------------------------------------------------------------------------- /xtask/src/commands/validate.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/commands/validate.rs -------------------------------------------------------------------------------- /xtask/src/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/main.rs -------------------------------------------------------------------------------- /xtask/src/utils/cargo.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/utils/cargo.rs -------------------------------------------------------------------------------- /xtask/src/utils/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/utils/mod.rs -------------------------------------------------------------------------------- /xtask/src/utils/process.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/utils/process.rs -------------------------------------------------------------------------------- /xtask/src/utils/prompt.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/utils/prompt.rs -------------------------------------------------------------------------------- /xtask/src/utils/time.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/utils/time.rs -------------------------------------------------------------------------------- /xtask/src/utils/workspace.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tracel-ai/cubecl/HEAD/xtask/src/utils/workspace.rs --------------------------------------------------------------------------------