Crash on call to DType.sizeof() when also creating a TensorShape [BUG]

[JakeEBrooks]

Bug description

Please forgive my ignorance... it's my first time submitting an issue on GitHub :)

Earlier today I found a weird crash that happens when TensorShape is called within a function that returns an object containing a DType. If I call sizeof on the DType field I get a compiler crash (see below).

Steps to reproduce

For me, running the following works fine:

struct SomeStruct:
    var datatype: DType

    fn __init__(inout self, datatype: DType):
        self.datatype = datatype

fn create_struct() -> SomeStruct:
    #var ax = TensorShape(0)
    var datatype = DType.uint32
    return SomeStruct(datatype)

fn main():
    var astruct = create_struct()
    print(astruct.datatype.sizeof())

But if I uncomment the TensorShape bit in create_struct, I get:

/__w/modular/modular/Kernels/mojo/stdlib/builtin/dtype.mojo:430:44: error: failed to legalize operation 'pop.dtype.sizeof' that was explicitly marked illegal
/__w/modular/modular/Kernels/mojo/stdlib/builtin/dtype.mojo:430:44: note: see current operation: %61 = "pop.dtype.sizeof"(%60) : (!kgen.dtype) -> index
mojo: error: Failed to materialize symbols: { (exec, { KGEN_EE_JIT_GlobalConstructor, KGEN_EE_JIT_GlobalDestructor, main }) } (from the layer: failed to lower module to LLVM IR for archive compilation)

Interestingly, I noticed if you just provide an empty TensorShape it works. For example, the following works fine for me:

struct SomeStruct:
    var datatype: DType

    fn __init__(inout self, datatype: DType):
        self.datatype = datatype

fn create_struct() -> SomeStruct:
    var ax = TensorShape()
    var datatype = DType.uint32
    return SomeStruct(datatype)

fn main():
    var astruct = create_struct()
    print(astruct.datatype.sizeof())

Hope I'm not missing anything, and that this is helpful!

System information

- What OS did you do install Mojo on ?
Ubuntu 23.10
- Provide version information for Mojo by pasting the output of `mojo -v`
mojo 24.2.0 (c2427bc5)
- Provide Modular CLI version by pasting the output of `modular -v`
modular 0.6.0 (04c05243)

[JakeEBrooks]

Reported fixed in #2983.