Skip to content

Add new devices to Habitat #21

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

Merged
merged 18 commits into from
Feb 6, 2023
Merged

Add new devices to Habitat #21

merged 18 commits into from
Feb 6, 2023

Conversation

jimgao1
Copy link
Contributor

@jimgao1 jimgao1 commented Feb 6, 2023

This PR adds support for the following devices to Habitat:

  • Desktop: RTX3090
  • Datacenter/Workstation: A100, A40, A4000, RTX4000

Copy link
Contributor

@johncalesp johncalesp left a comment

Choose a reason for hiding this comment

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

Looks great!, once it's merged, I will add the new card's information to the frontend

Copy link
Contributor

@michaelshin michaelshin left a comment

Choose a reason for hiding this comment

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

Looks good just added minor feedback

print("Generating dataset")
# generate vectorized dataset (one entry for each device with device params)
device_params = get_all_devices(device_features)

x, y = [], []
for device in devices.keys():
df_merged_device = df_merged[features + [device, ]]
for row in tqdm(df_merged_device.iterrows(), leave=False, desc=device, total=len(df_merged_device.index)):
# df_merged_device = df_merged[features + [device, ]]
Copy link
Contributor

Choose a reason for hiding this comment

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

you can remove this comment

habitat.Device.A100,
habitat.Device.T4,
habitat.Device.RTX2070,
# habitat.Device.P4000,
Copy link
Contributor

Choose a reason for hiding this comment

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

You can remove the commented devices

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Oh good catch. I actually didn't mean to remove it.

@jimgao1 jimgao1 merged commit fb0209d into main Feb 6, 2023
@jimgao1 jimgao1 deleted the dec22-new-devices branch February 6, 2023 19:39
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.

3 participants