diff --git a/python/cutlass/__init__.py b/python/cutlass_cppgen/__init__.py similarity index 100% rename from python/cutlass/__init__.py rename to python/cutlass_cppgen/__init__.py diff --git a/python/cutlass/backend/__init__.py b/python/cutlass_cppgen/backend/__init__.py similarity index 100% rename from python/cutlass/backend/__init__.py rename to python/cutlass_cppgen/backend/__init__.py diff --git a/python/cutlass/backend/arguments.py b/python/cutlass_cppgen/backend/arguments.py similarity index 100% rename from python/cutlass/backend/arguments.py rename to python/cutlass_cppgen/backend/arguments.py diff --git a/python/cutlass/backend/c_types.py b/python/cutlass_cppgen/backend/c_types.py similarity index 100% rename from python/cutlass/backend/c_types.py rename to python/cutlass_cppgen/backend/c_types.py diff --git a/python/cutlass/backend/compiler.py b/python/cutlass_cppgen/backend/compiler.py similarity index 100% rename from python/cutlass/backend/compiler.py rename to python/cutlass_cppgen/backend/compiler.py diff --git a/python/cutlass/backend/conv2d_operation.py b/python/cutlass_cppgen/backend/conv2d_operation.py similarity index 100% rename from python/cutlass/backend/conv2d_operation.py rename to python/cutlass_cppgen/backend/conv2d_operation.py diff --git a/python/cutlass/backend/epilogue.py b/python/cutlass_cppgen/backend/epilogue.py similarity index 100% rename from python/cutlass/backend/epilogue.py rename to python/cutlass_cppgen/backend/epilogue.py diff --git a/python/cutlass/backend/evt/__init__.py b/python/cutlass_cppgen/backend/evt/__init__.py similarity index 100% rename from python/cutlass/backend/evt/__init__.py rename to python/cutlass_cppgen/backend/evt/__init__.py diff --git a/python/cutlass/backend/evt/backend/__init__.py b/python/cutlass_cppgen/backend/evt/backend/__init__.py similarity index 100% rename from python/cutlass/backend/evt/backend/__init__.py rename to python/cutlass_cppgen/backend/evt/backend/__init__.py diff --git a/python/cutlass/backend/evt/backend/emitter_base.py b/python/cutlass_cppgen/backend/evt/backend/emitter_base.py similarity index 100% rename from python/cutlass/backend/evt/backend/emitter_base.py rename to python/cutlass_cppgen/backend/evt/backend/emitter_base.py diff --git a/python/cutlass/backend/evt/backend/sm100_emitter.py b/python/cutlass_cppgen/backend/evt/backend/sm100_emitter.py similarity index 100% rename from python/cutlass/backend/evt/backend/sm100_emitter.py rename to python/cutlass_cppgen/backend/evt/backend/sm100_emitter.py diff --git a/python/cutlass/backend/evt/backend/sm100_nodes.py b/python/cutlass_cppgen/backend/evt/backend/sm100_nodes.py similarity index 100% rename from python/cutlass/backend/evt/backend/sm100_nodes.py rename to python/cutlass_cppgen/backend/evt/backend/sm100_nodes.py diff --git a/python/cutlass/backend/evt/backend/sm80_emitter.py b/python/cutlass_cppgen/backend/evt/backend/sm80_emitter.py similarity index 100% rename from python/cutlass/backend/evt/backend/sm80_emitter.py rename to python/cutlass_cppgen/backend/evt/backend/sm80_emitter.py diff --git a/python/cutlass/backend/evt/backend/sm80_nodes.py b/python/cutlass_cppgen/backend/evt/backend/sm80_nodes.py similarity index 100% rename from python/cutlass/backend/evt/backend/sm80_nodes.py rename to python/cutlass_cppgen/backend/evt/backend/sm80_nodes.py diff --git a/python/cutlass/backend/evt/backend/sm90_emitter.py b/python/cutlass_cppgen/backend/evt/backend/sm90_emitter.py similarity index 100% rename from python/cutlass/backend/evt/backend/sm90_emitter.py rename to python/cutlass_cppgen/backend/evt/backend/sm90_emitter.py diff --git a/python/cutlass/backend/evt/backend/sm90_nodes.py b/python/cutlass_cppgen/backend/evt/backend/sm90_nodes.py similarity index 100% rename from python/cutlass/backend/evt/backend/sm90_nodes.py rename to python/cutlass_cppgen/backend/evt/backend/sm90_nodes.py diff --git a/python/cutlass/backend/evt/epilogue.py b/python/cutlass_cppgen/backend/evt/epilogue.py similarity index 100% rename from python/cutlass/backend/evt/epilogue.py rename to python/cutlass_cppgen/backend/evt/epilogue.py diff --git a/python/cutlass/backend/evt/frontend/__init__.py b/python/cutlass_cppgen/backend/evt/frontend/__init__.py similarity index 100% rename from python/cutlass/backend/evt/frontend/__init__.py rename to python/cutlass_cppgen/backend/evt/frontend/__init__.py diff --git a/python/cutlass/backend/evt/frontend/frontend_base.py b/python/cutlass_cppgen/backend/evt/frontend/frontend_base.py similarity index 100% rename from python/cutlass/backend/evt/frontend/frontend_base.py rename to python/cutlass_cppgen/backend/evt/frontend/frontend_base.py diff --git a/python/cutlass/backend/evt/frontend/python_ast.py b/python/cutlass_cppgen/backend/evt/frontend/python_ast.py similarity index 100% rename from python/cutlass/backend/evt/frontend/python_ast.py rename to python/cutlass_cppgen/backend/evt/frontend/python_ast.py diff --git a/python/cutlass/backend/evt/ir/__init__.py b/python/cutlass_cppgen/backend/evt/ir/__init__.py similarity index 100% rename from python/cutlass/backend/evt/ir/__init__.py rename to python/cutlass_cppgen/backend/evt/ir/__init__.py diff --git a/python/cutlass/backend/evt/ir/compute_nodes.py b/python/cutlass_cppgen/backend/evt/ir/compute_nodes.py similarity index 100% rename from python/cutlass/backend/evt/ir/compute_nodes.py rename to python/cutlass_cppgen/backend/evt/ir/compute_nodes.py diff --git a/python/cutlass/backend/evt/ir/dag_ir.py b/python/cutlass_cppgen/backend/evt/ir/dag_ir.py similarity index 100% rename from python/cutlass/backend/evt/ir/dag_ir.py rename to python/cutlass_cppgen/backend/evt/ir/dag_ir.py diff --git a/python/cutlass/backend/evt/ir/layout_algorithm.py b/python/cutlass_cppgen/backend/evt/ir/layout_algorithm.py similarity index 100% rename from python/cutlass/backend/evt/ir/layout_algorithm.py rename to python/cutlass_cppgen/backend/evt/ir/layout_algorithm.py diff --git a/python/cutlass/backend/evt/ir/layout_nodes.py b/python/cutlass_cppgen/backend/evt/ir/layout_nodes.py similarity index 100% rename from python/cutlass/backend/evt/ir/layout_nodes.py rename to python/cutlass_cppgen/backend/evt/ir/layout_nodes.py diff --git a/python/cutlass/backend/evt/ir/load_nodes.py b/python/cutlass_cppgen/backend/evt/ir/load_nodes.py similarity index 100% rename from python/cutlass/backend/evt/ir/load_nodes.py rename to python/cutlass_cppgen/backend/evt/ir/load_nodes.py diff --git a/python/cutlass/backend/evt/ir/node.py b/python/cutlass_cppgen/backend/evt/ir/node.py similarity index 100% rename from python/cutlass/backend/evt/ir/node.py rename to python/cutlass_cppgen/backend/evt/ir/node.py diff --git a/python/cutlass/backend/evt/ir/store_nodes.py b/python/cutlass_cppgen/backend/evt/ir/store_nodes.py similarity index 100% rename from python/cutlass/backend/evt/ir/store_nodes.py rename to python/cutlass_cppgen/backend/evt/ir/store_nodes.py diff --git a/python/cutlass/backend/evt/ir/tensor.py b/python/cutlass_cppgen/backend/evt/ir/tensor.py similarity index 100% rename from python/cutlass/backend/evt/ir/tensor.py rename to python/cutlass_cppgen/backend/evt/ir/tensor.py diff --git a/python/cutlass/backend/evt/passes/__init__.py b/python/cutlass_cppgen/backend/evt/passes/__init__.py similarity index 100% rename from python/cutlass/backend/evt/passes/__init__.py rename to python/cutlass_cppgen/backend/evt/passes/__init__.py diff --git a/python/cutlass/backend/evt/passes/graph_drawer.py b/python/cutlass_cppgen/backend/evt/passes/graph_drawer.py similarity index 100% rename from python/cutlass/backend/evt/passes/graph_drawer.py rename to python/cutlass_cppgen/backend/evt/passes/graph_drawer.py diff --git a/python/cutlass/backend/evt/passes/pass_argument_type.py b/python/cutlass_cppgen/backend/evt/passes/pass_argument_type.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_argument_type.py rename to python/cutlass_cppgen/backend/evt/passes/pass_argument_type.py diff --git a/python/cutlass/backend/evt/passes/pass_dag_2_tree.py b/python/cutlass_cppgen/backend/evt/passes/pass_dag_2_tree.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_dag_2_tree.py rename to python/cutlass_cppgen/backend/evt/passes/pass_dag_2_tree.py diff --git a/python/cutlass/backend/evt/passes/pass_fix_element_d.py b/python/cutlass_cppgen/backend/evt/passes/pass_fix_element_d.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_fix_element_d.py rename to python/cutlass_cppgen/backend/evt/passes/pass_fix_element_d.py diff --git a/python/cutlass/backend/evt/passes/pass_get_impl.py b/python/cutlass_cppgen/backend/evt/passes/pass_get_impl.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_get_impl.py rename to python/cutlass_cppgen/backend/evt/passes/pass_get_impl.py diff --git a/python/cutlass/backend/evt/passes/pass_layout_elimination.py b/python/cutlass_cppgen/backend/evt/passes/pass_layout_elimination.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_layout_elimination.py rename to python/cutlass_cppgen/backend/evt/passes/pass_layout_elimination.py diff --git a/python/cutlass/backend/evt/passes/pass_manager.py b/python/cutlass_cppgen/backend/evt/passes/pass_manager.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_manager.py rename to python/cutlass_cppgen/backend/evt/passes/pass_manager.py diff --git a/python/cutlass/backend/evt/passes/pass_no_op_elimination.py b/python/cutlass_cppgen/backend/evt/passes/pass_no_op_elimination.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_no_op_elimination.py rename to python/cutlass_cppgen/backend/evt/passes/pass_no_op_elimination.py diff --git a/python/cutlass/backend/evt/passes/pass_preprocess_red.py b/python/cutlass_cppgen/backend/evt/passes/pass_preprocess_red.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_preprocess_red.py rename to python/cutlass_cppgen/backend/evt/passes/pass_preprocess_red.py diff --git a/python/cutlass/backend/evt/passes/pass_shape_type_propagation.py b/python/cutlass_cppgen/backend/evt/passes/pass_shape_type_propagation.py similarity index 100% rename from python/cutlass/backend/evt/passes/pass_shape_type_propagation.py rename to python/cutlass_cppgen/backend/evt/passes/pass_shape_type_propagation.py diff --git a/python/cutlass/backend/evt/passes/smem_size_calculator.py b/python/cutlass_cppgen/backend/evt/passes/smem_size_calculator.py similarity index 100% rename from python/cutlass/backend/evt/passes/smem_size_calculator.py rename to python/cutlass_cppgen/backend/evt/passes/smem_size_calculator.py diff --git a/python/cutlass/backend/evt/passes/util.py b/python/cutlass_cppgen/backend/evt/passes/util.py similarity index 100% rename from python/cutlass/backend/evt/passes/util.py rename to python/cutlass_cppgen/backend/evt/passes/util.py diff --git a/python/cutlass/backend/frontend.py b/python/cutlass_cppgen/backend/frontend.py similarity index 100% rename from python/cutlass/backend/frontend.py rename to python/cutlass_cppgen/backend/frontend.py diff --git a/python/cutlass/backend/gemm_operation.py b/python/cutlass_cppgen/backend/gemm_operation.py similarity index 100% rename from python/cutlass/backend/gemm_operation.py rename to python/cutlass_cppgen/backend/gemm_operation.py diff --git a/python/cutlass/backend/library.py b/python/cutlass_cppgen/backend/library.py similarity index 100% rename from python/cutlass/backend/library.py rename to python/cutlass_cppgen/backend/library.py diff --git a/python/cutlass/backend/memory_manager.py b/python/cutlass_cppgen/backend/memory_manager.py similarity index 100% rename from python/cutlass/backend/memory_manager.py rename to python/cutlass_cppgen/backend/memory_manager.py diff --git a/python/cutlass/backend/operation.py b/python/cutlass_cppgen/backend/operation.py similarity index 100% rename from python/cutlass/backend/operation.py rename to python/cutlass_cppgen/backend/operation.py diff --git a/python/cutlass/backend/reduction_operation.py b/python/cutlass_cppgen/backend/reduction_operation.py similarity index 100% rename from python/cutlass/backend/reduction_operation.py rename to python/cutlass_cppgen/backend/reduction_operation.py diff --git a/python/cutlass/backend/type_hint.py b/python/cutlass_cppgen/backend/type_hint.py similarity index 100% rename from python/cutlass/backend/type_hint.py rename to python/cutlass_cppgen/backend/type_hint.py diff --git a/python/cutlass/backend/utils/__init__.py b/python/cutlass_cppgen/backend/utils/__init__.py similarity index 100% rename from python/cutlass/backend/utils/__init__.py rename to python/cutlass_cppgen/backend/utils/__init__.py diff --git a/python/cutlass/backend/utils/device.py b/python/cutlass_cppgen/backend/utils/device.py similarity index 100% rename from python/cutlass/backend/utils/device.py rename to python/cutlass_cppgen/backend/utils/device.py diff --git a/python/cutlass/emit/__init__.py b/python/cutlass_cppgen/emit/__init__.py similarity index 100% rename from python/cutlass/emit/__init__.py rename to python/cutlass_cppgen/emit/__init__.py diff --git a/python/cutlass/emit/common.py b/python/cutlass_cppgen/emit/common.py similarity index 100% rename from python/cutlass/emit/common.py rename to python/cutlass_cppgen/emit/common.py diff --git a/python/cutlass/emit/pytorch.py b/python/cutlass_cppgen/emit/pytorch.py similarity index 100% rename from python/cutlass/emit/pytorch.py rename to python/cutlass_cppgen/emit/pytorch.py diff --git a/python/cutlass/epilogue/__init__.py b/python/cutlass_cppgen/epilogue/__init__.py similarity index 100% rename from python/cutlass/epilogue/__init__.py rename to python/cutlass_cppgen/epilogue/__init__.py diff --git a/python/cutlass/epilogue/epilogue.py b/python/cutlass_cppgen/epilogue/epilogue.py similarity index 100% rename from python/cutlass/epilogue/epilogue.py rename to python/cutlass_cppgen/epilogue/epilogue.py diff --git a/python/cutlass/epilogue/evt_ops.py b/python/cutlass_cppgen/epilogue/evt_ops.py similarity index 100% rename from python/cutlass/epilogue/evt_ops.py rename to python/cutlass_cppgen/epilogue/evt_ops.py diff --git a/python/cutlass/library_defaults.py b/python/cutlass_cppgen/library_defaults.py similarity index 100% rename from python/cutlass/library_defaults.py rename to python/cutlass_cppgen/library_defaults.py diff --git a/python/cutlass/op/__init__.py b/python/cutlass_cppgen/op/__init__.py similarity index 100% rename from python/cutlass/op/__init__.py rename to python/cutlass_cppgen/op/__init__.py diff --git a/python/cutlass/op/conv.py b/python/cutlass_cppgen/op/conv.py similarity index 100% rename from python/cutlass/op/conv.py rename to python/cutlass_cppgen/op/conv.py diff --git a/python/cutlass/op/gemm.py b/python/cutlass_cppgen/op/gemm.py similarity index 100% rename from python/cutlass/op/gemm.py rename to python/cutlass_cppgen/op/gemm.py diff --git a/python/cutlass/op/gemm_grouped.py b/python/cutlass_cppgen/op/gemm_grouped.py similarity index 100% rename from python/cutlass/op/gemm_grouped.py rename to python/cutlass_cppgen/op/gemm_grouped.py diff --git a/python/cutlass/op/op.py b/python/cutlass_cppgen/op/op.py similarity index 100% rename from python/cutlass/op/op.py rename to python/cutlass_cppgen/op/op.py diff --git a/python/cutlass/shape.py b/python/cutlass_cppgen/shape.py similarity index 100% rename from python/cutlass/shape.py rename to python/cutlass_cppgen/shape.py diff --git a/python/cutlass/swizzle.py b/python/cutlass_cppgen/swizzle.py similarity index 100% rename from python/cutlass/swizzle.py rename to python/cutlass_cppgen/swizzle.py diff --git a/python/cutlass/utils/__init__.py b/python/cutlass_cppgen/utils/__init__.py similarity index 100% rename from python/cutlass/utils/__init__.py rename to python/cutlass_cppgen/utils/__init__.py diff --git a/python/cutlass/utils/check.py b/python/cutlass_cppgen/utils/check.py similarity index 100% rename from python/cutlass/utils/check.py rename to python/cutlass_cppgen/utils/check.py diff --git a/python/cutlass/utils/datatypes.py b/python/cutlass_cppgen/utils/datatypes.py similarity index 100% rename from python/cutlass/utils/datatypes.py rename to python/cutlass_cppgen/utils/datatypes.py diff --git a/python/cutlass/utils/lazy_import.py b/python/cutlass_cppgen/utils/lazy_import.py similarity index 100% rename from python/cutlass/utils/lazy_import.py rename to python/cutlass_cppgen/utils/lazy_import.py diff --git a/python/cutlass/utils/profiler.py b/python/cutlass_cppgen/utils/profiler.py similarity index 100% rename from python/cutlass/utils/profiler.py rename to python/cutlass_cppgen/utils/profiler.py diff --git a/setup.cfg b/setup.cfg index d2394fe6..ee3d1831 100644 --- a/setup.cfg +++ b/setup.cfg @@ -20,7 +20,7 @@ packages = cutlass_library.source pycute package_dir = - cutlass_cppgen=python/cutlass + cutlass_cppgen=python/cutlass_cppgen cutlass_library=python/cutlass_library cutlass_library.source=. pycute=python/pycute