@@ -437,31 +437,33 @@ MPS_EXECUTOR_RUNNER_SRCS = [
437437 "extension/data_loader/file_data_loader.cpp" ,
438438]
439439
440- MPS_BACKEND_SRCS = [
441- "backends/apple/mps/runtime/MPSBackend.mm" ,
442- "backends/apple/mps/runtime/MPSCompiler.mm" ,
443- "backends/apple/mps/runtime/MPSDelegateHeader.mm" ,
444- "backends/apple/mps/runtime/MPSDevice.mm" ,
445- "backends/apple/mps/runtime/MPSExecutor.mm" ,
446- "backends/apple/mps/runtime/MPSGraphBuilder.mm" ,
447- "backends/apple/mps/runtime/MPSStream.mm" ,
448- "backends/apple/mps/runtime/operations/ActivationOps.mm" ,
449- "backends/apple/mps/runtime/operations/BinaryOps.mm" ,
450- "backends/apple/mps/runtime/operations/ClampOps.mm" ,
451- "backends/apple/mps/runtime/operations/ConstantOps.mm" ,
452- "backends/apple/mps/runtime/operations/ConvolutionOps.mm" ,
453- "backends/apple/mps/runtime/operations/IndexingOps.mm" ,
454- "backends/apple/mps/runtime/operations/LinearAlgebra.mm" ,
455- "backends/apple/mps/runtime/operations/NormalizationOps.mm" ,
456- "backends/apple/mps/runtime/operations/OperationUtils.mm" ,
457- "backends/apple/mps/runtime/operations/PadOps.mm" ,
458- "backends/apple/mps/runtime/operations/PoolingOps.mm" ,
459- "backends/apple/mps/runtime/operations/QuantDequant.mm" ,
460- "backends/apple/mps/runtime/operations/RangeOps.mm" ,
461- "backends/apple/mps/runtime/operations/ReduceOps.mm" ,
462- "backends/apple/mps/runtime/operations/ShapeOps.mm" ,
463- "backends/apple/mps/runtime/operations/UnaryOps.mm" ,
464- ]
440+ MPS_BACKEND_BUCK_SRCS = [
441+ "runtime/MPSBackend.mm" ,
442+ "runtime/MPSCompiler.mm" ,
443+ "runtime/MPSDelegateHeader.mm" ,
444+ "runtime/MPSDevice.mm" ,
445+ "runtime/MPSExecutor.mm" ,
446+ "runtime/MPSGraphBuilder.mm" ,
447+ "runtime/MPSStream.mm" ,
448+ "runtime/operations/ActivationOps.mm" ,
449+ "runtime/operations/BinaryOps.mm" ,
450+ "runtime/operations/ClampOps.mm" ,
451+ "runtime/operations/ConstantOps.mm" ,
452+ "runtime/operations/ConvolutionOps.mm" ,
453+ "runtime/operations/IndexingOps.mm" ,
454+ "runtime/operations/LinearAlgebra.mm" ,
455+ "runtime/operations/NormalizationOps.mm" ,
456+ "runtime/operations/OperationUtils.mm" ,
457+ "runtime/operations/PadOps.mm" ,
458+ "runtime/operations/PoolingOps.mm" ,
459+ "runtime/operations/QuantDequant.mm" ,
460+ "runtime/operations/RangeOps.mm" ,
461+ "runtime/operations/ReduceOps.mm" ,
462+ "runtime/operations/ShapeOps.mm" ,
463+ "runtime/operations/UnaryOps.mm" ,
464+ ]
465+
466+ MPS_BACKEND_SRCS = ["backends/apple/mps/" + x for x in MPS_BACKEND_BUCK_SRCS ]
465467
466468MPS_SCHEMA_SRCS = [
467469 "backends/apple/mps/serialization/schema.fbs" ,
0 commit comments