#include <metal_stdlib>

using namespace metal;
float3 f(int i) {
  float3x3 m = float3x3(float3(1.0f, 2.0f, 3.0f), float3(4.0f, 5.0f, 6.0f), float3(7.0f, 8.0f, 9.0f));
  return m[i];
}