Add support for other same size vector bitcasts to Mono (#104309)

* Add support for other same size vector bitcasts to Mono

* Ensure get_common_simd_info handles non-generic input parameters

* Ensure emit_sri_vector128 correctly checks for the generic type argument

src/mono/mono/mini/interp/simd-methods.def

@@ -31,11 +31,16 @@ SIMD_METHOD(AsInt32)
 SIMD_METHOD(AsInt64)
 SIMD_METHOD(AsNInt)
 SIMD_METHOD(AsNUInt)
+SIMD_METHOD(AsPlane)
+SIMD_METHOD(AsQuaternion)
 SIMD_METHOD(AsSByte)
 SIMD_METHOD(AsSingle)
 SIMD_METHOD(AsUInt16)
 SIMD_METHOD(AsUInt32)
 SIMD_METHOD(AsUInt64)
+SIMD_METHOD(AsVector)
+SIMD_METHOD(AsVector4)
+SIMD_METHOD(AsVector128)
 SIMD_METHOD(ConditionalSelect)
 SIMD_METHOD(Create)
 SIMD_METHOD(CreateScalar)

src/mono/mono/mini/interp/transform-simd.c

@@ -64,11 +64,16 @@ static guint16 sri_vector128_methods [] = {
 	SN_AsInt64,
 	SN_AsNInt,
 	SN_AsNUInt,
+	SN_AsPlane,
+	SN_AsQuaternion,
 	SN_AsSByte,
 	SN_AsSingle,
 	SN_AsUInt16,
 	SN_AsUInt32,
 	SN_AsUInt64,
+	SN_AsVector,
+	SN_AsVector4,
+	SN_AsVector128,
 	SN_ConditionalSelect,
 	SN_Create,
 	SN_CreateScalar,
@@ -424,7 +429,13 @@ emit_sri_vector128 (TransformData *td, MonoMethod *cmethod, MonoMethodSignature
 	gint16 simd_opcode = -1;
 	gint16 simd_intrins = -1;
 
-	vector_klass = mono_class_from_mono_type_internal (csignature->ret);
+	if (csignature->ret->type == MONO_TYPE_GENERICINST) {
+		vector_klass = mono_class_from_mono_type_internal (csignature->ret);
+	} else if (csignature->params [0]->type == MONO_TYPE_GENERICINST) {
+		vector_klass = mono_class_from_mono_type_internal (csignature->params [0]);
+	} else {
+		return FALSE;
+	}
 
 	MonoTypeEnum atype;
 	int arg_size, scalar_arg;
@@ -444,6 +455,8 @@ emit_sri_vector128 (TransformData *td, MonoMethod *cmethod, MonoMethodSignature
 		case SN_AsInt64:
 		case SN_AsNInt:
 		case SN_AsNUInt:
+		case SN_AsPlane:
+		case SN_AsQuaternion:
 		case SN_AsSByte:
 		case SN_AsSingle:
 		case SN_AsUInt16:
@@ -455,6 +468,25 @@ emit_sri_vector128 (TransformData *td, MonoMethod *cmethod, MonoMethodSignature
 			simd_intrins = INTERP_SIMD_INTRINSIC_V128_BITCAST;
 			break;
 		}
+		case SN_AsVector:
+		case SN_AsVector128:
+		case SN_AsVector4: {
+			if (!is_element_type_primitive (csignature->ret) || !is_element_type_primitive (csignature->params [0]))
+				return FALSE;
+
+			MonoClass *ret_class = mono_class_from_mono_type_internal (csignature->ret);
+			int ret_size = mono_class_value_size (ret_class, NULL);
+
+			MonoClass *arg_class = mono_class_from_mono_type_internal (csignature->params [0]);
+			int arg_size = mono_class_value_size (arg_class, NULL);
+
+			if (arg_size == ret_size) {
+				simd_opcode = MINT_SIMD_INTRINS_P_P;
+				simd_intrins = INTERP_SIMD_INTRINSIC_V128_BITCAST;
+				break;
+			}
+			return FALSE;
+		}
 		case SN_ConditionalSelect:
 			simd_opcode = MINT_SIMD_INTRINS_P_PPP;
 			simd_intrins = INTERP_SIMD_INTRINSIC_V128_CONDITIONAL_SELECT;

src/mono/mono/mini/simd-intrinsics.c

@@ -1187,11 +1187,16 @@ static guint16 sri_vector_methods [] = {
 	SN_AsInt64,
 	SN_AsNInt,
 	SN_AsNUInt,
+	SN_AsPlane,
+	SN_AsQuaternion,
 	SN_AsSByte,
 	SN_AsSingle,
 	SN_AsUInt16,
 	SN_AsUInt32,
 	SN_AsUInt64,
+	SN_AsVector,
+	SN_AsVector128,
+	SN_AsVector4,
 	SN_BitwiseAnd,
 	SN_BitwiseOr,
 	SN_Ceiling,
@@ -1622,6 +1627,8 @@ emit_sri_vector (MonoCompile *cfg, MonoMethod *cmethod, MonoMethodSignature *fsi
 	case SN_AsInt64:
 	case SN_AsNInt:
 	case SN_AsNUInt:
+	case SN_AsPlane:
+	case SN_AsQuaternion:
 	case SN_AsSByte:
 	case SN_AsSingle:
 	case SN_AsUInt16:
@@ -1631,6 +1638,23 @@ emit_sri_vector (MonoCompile *cfg, MonoMethod *cmethod, MonoMethodSignature *fsi
 			return NULL;
 		return emit_simd_ins (cfg, klass, OP_XCAST, args [0]->dreg, -1);
 	}
+	case SN_AsVector:
+	case SN_AsVector128:
+	case SN_AsVector4: {
+		if (!is_element_type_primitive (fsig->ret) || !is_element_type_primitive (fsig->params [0]))
+			return NULL;
+
+		MonoClass *ret_class = mono_class_from_mono_type_internal (fsig->ret);
+		int ret_size = mono_class_value_size (ret_class, NULL);
+
+		MonoClass *arg_class = mono_class_from_mono_type_internal (fsig->params [0]);
+		int arg_size = mono_class_value_size (arg_class, NULL);
+
+		if (arg_size == ret_size)
+			return emit_simd_ins (cfg, klass, OP_XCAST, args [0]->dreg, -1);
+
+		return NULL;
+	}
 	case SN_Ceiling:
 	case SN_Floor: {
 		if (!type_enum_is_float (arg0_type))

src/mono/mono/mini/simd-methods.h

@@ -73,11 +73,14 @@ METHOD(AsInt32)
 METHOD(AsInt64)
 METHOD(AsNInt)
 METHOD(AsNUInt)
+METHOD(AsPlane)
+METHOD(AsQuaternion)
 METHOD(AsSByte)
 METHOD(AsSingle)
 METHOD(AsUInt16)
 METHOD(AsUInt32)
 METHOD(AsUInt64)
+METHOD(AsVector)
 METHOD(AsVector128)
 METHOD(AsVector2)
 METHOD(AsVector256)