(
op: CumOpType, shape: number[], exclusive: boolean, reverse: boolean)
| 38 | op: CumOpType; |
| 39 | |
| 40 | constructor( |
| 41 | op: CumOpType, shape: number[], exclusive: boolean, reverse: boolean) { |
| 42 | this.workgroupSize = [128, 1, 1]; |
| 43 | this.outputShape = shape; |
| 44 | this.dispatchLayout = flatDispatchLayout(this.outputShape); |
| 45 | this.dispatch = computeDispatch( |
| 46 | this.dispatchLayout, this.outputShape, this.workgroupSize); |
| 47 | this.exclusive = exclusive; |
| 48 | this.reverse = reverse; |
| 49 | this.op = op; |
| 50 | this.shaderKey = `cum_${this.op}_${this.exclusive}_${this.reverse}`; |
| 51 | } |
| 52 | |
| 53 | getUserCode(): string { |
| 54 | const rank = this.outputShape.length; |
nothing calls this directly
no test coverage detected