(&self, a: &Tensor, b: &Tensor, c: &Tensor)
| 1310 | } |
| 1311 | |
| 1312 | fn add3(&self, a: &Tensor, b: &Tensor, c: &Tensor) -> Result<Tensor> { |
| 1313 | let n = a.elem_count(); |
| 1314 | if n > 32768 { |
| 1315 | log::debug!("add3 GPU: {n} elements"); |
| 1316 | self.dispatch_ternary_vec4(a, b, c, "add3") |
| 1317 | } else { |
| 1318 | log::debug!("add3 CPU: {n} elements (<=32768)"); |
| 1319 | ((a + b)? + c)?.contiguous() |
| 1320 | } |
| 1321 | } |
| 1322 | |
| 1323 | fn exp_mul(&self, x: &Tensor, y: &Tensor) -> Result<Tensor> { |
| 1324 | let n = x.elem_count(); |
nothing calls this directly
no test coverage detected