| 293 | } |
| 294 | |
| 295 | void setN(uint64_t n) |
| 296 | { |
| 297 | PLUGIN_CUBLASASSERT(cublasLtMatrixLayoutSetAttribute(Bdesc, CUBLASLT_MATRIX_LAYOUT_COLS, &n, sizeof(n))); |
| 298 | PLUGIN_CUBLASASSERT(cublasLtMatrixLayoutSetAttribute(Cdesc, CUBLASLT_MATRIX_LAYOUT_COLS, &n, sizeof(n))); |
| 299 | } |
| 300 | }; |
| 301 | |
| 302 | template <typename T> |