THCTensor_getDevicestate tensor实现分析
THCTensor_(getDevice)函数是THC库中的一个函数,用于获取THC张量的设备编号。其实现如下:
int THCTensor_(getDevice)(THCState *state, const THCTensor *tensor) {
THAssert(THCudaTensor_checkGPU(state, 1, tensor));
return THCudaTensor_getDevice(state, tensor);
}
该函数首先通过THCudaTensor_checkGPU函数检查输入的张量是否在GPU上,如果不在则会触发一个错误。然后,它调用THCudaTensor_getDevice函数获取张量的设备编号,并将其返回。
总之,THCTensor_(getDevice)函数的作用是获取THC张量的设备编号。
原文地址: https://www.cveoy.top/t/topic/e13D 著作权归作者所有。请勿转载和采集!