|
发表于 2023-9-13 14:37:42
|
显示全部楼层
我的显卡是4090,请问出现这个报错怎么解决
Failed inference for CodeFormer: nvrtc: error: invalid value for --gpu-architecture (-arch)
nvrtc compilation failed:
#define NAN int_as_float(0x7fffffff)
#define POS_INFINITY __int_as_float(0x7f800000)
#define NEG_INFINITY __int_as_float(0xff800000)
template<typename T>
__device T maximum(T a, T b) {
return isnan(a) ? a : (a > b ? a : b);
}
template<typename T>
device T minimum(T a, T b) {
return isnan(a) ? a : (a < b ? a : b);
}
extern "C" global
void fused_sigmoid_mul(float* tx_1, float* aten_mul) {
{
float tx_1_1 = __ldg(tx_1 + (512 * blockIdx.x + threadIdx.x) % 16777216);
aten_mul[512 * blockIdx.x + threadIdx.x] = tx_1_1 * (1.f / (1.f + (expf(0.f - tx_1_1))));
}
} |
|