@@ -437,31 +437,33 @@ MPS_EXECUTOR_RUNNER_SRCS = [
437
437
"extension/data_loader/file_data_loader.cpp" ,
438
438
]
439
439
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 ]
465
467
466
468
MPS_SCHEMA_SRCS = [
467
469
"backends/apple/mps/serialization/schema.fbs" ,
0 commit comments