Method set

src/utils/miopen.cpp:211–263

}

void TensorDescriptor::set(miopenDataType_t data_type,
                           std::vector<int> dims_in,
                           std::vector<int> strides_in)
{

  // Check that arguments are valid
  if (dims_in.empty()) {
    LBANN_ERROR("attempted to set MIOpen tensor descriptor with no dimensions");
  }
  if (!strides_in.empty() && dims_in.size() != strides_in.size()) {
    LBANN_ERROR("attempted to set MIOpen tensor descriptor ",
                "with mismatched dimensions (",
                dims_in.size(),
                ") ",
                "and strides (",
                strides_in.size(),
                ")");
  }

  std::vector<int> dims = std::move(dims_in), strides = std::move(strides_in);
  if (dims.size() < 4) {
    switch (dims.size()) {
    case 2:
      dims = {dims[0], 1, dims[1], 1};
      strides = {};
      break;
    case 3:
      dims = {dims[0], 1, dims[1], dims[2]};
      strides = {};
      break;
    default:
      LBANN_ERROR("Dims of size 1. Don't know what to do.");
      break;
    }
  }

  // Assume data is contiguous if no strides are provided
  // Note (trb 12/29/2020): MIOpen only accepts contiguous strides.
  if (strides.empty()) {
    strides.resize(dims.size(), 1);
    for (int i = strides.size() - 1; i > 0; --i) {
      strides[i - 1] = strides[i] * dims[i];
    }
  }

  // Set MIOpen object
  create();
  CHECK_MIOPEN(miopenSetTensorDescriptor(desc_,
                                         data_type,
                                         dims.size(),
                                         dims.data(),
                                         strides.data()));
}

// -----------------------------
// DropoutDescriptor