Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Updating Metal version and removing data type overrides from MNIST model #52

Merged
merged 1 commit into from
Aug 9, 2024

Conversation

sdjordjevicTT
Copy link
Contributor

@sdjordjevicTT sdjordjevicTT commented Aug 9, 2024

We had to make workarounds for the MNIST model because the current version of Metal did not support the float32 data type. TT-MLIR has updated the version of Metal to support the float32 data types, so we are updating the reference to the TT-MLIR to include that Metal version update. Along the way, the bfloat16 overrides from the MNIST model will be removed to verify that everything is working as expected.

Solves: https://github.com/tenstorrent/tt-forge/issues/51

Copy link
Contributor

@vladimirjovanovicTT vladimirjovanovicTT left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

from pybuda._C import DataFormat
from pybuda.config import _get_global_compiler_config

I guess you don't need these anymore

@sdjordjevicTT sdjordjevicTT force-pushed the sdjordjevic/update_tt_mlir_and_metal_version branch from 8b7a37b to f24da15 Compare August 9, 2024 11:32
@sdjordjevicTT sdjordjevicTT force-pushed the sdjordjevic/update_tt_mlir_and_metal_version branch from f24da15 to d982513 Compare August 9, 2024 11:33
@sdjordjevicTT sdjordjevicTT merged commit 6d34c6b into main Aug 9, 2024
1 check failed
@sdjordjevicTT sdjordjevicTT deleted the sdjordjevic/update_tt_mlir_and_metal_version branch September 5, 2024 09:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants